from kedro.framework.session import KedroSession
from kedro_mlflow.config import get_mlflow_config

with KedroSession.create() as session:
  mlflow_config=get_mlflow_config()
  mlflow_config.setup()
  < your code>