fix deactivating

Galileo-Galilei · Dec 13, 2024 · 7d2c716 · 7d2c716
1 parent 7daeade
commit 7d2c716
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/kedro_mlflow/framework/hooks/mlflow_hook.py b/kedro_mlflow/framework/hooks/mlflow_hook.py
@@ -273,14 +273,15 @@ def before_pipeline_run(
                     pipeline_name=run_params["pipeline_name"],
                 ),
             )
+
+            add_run_id_to_artifact_datasets(catalog, mlflow.active_run().info.run_id)
+
         else:
             self._logger.info(
                 "kedro-mlflow logging is deactivated for this pipeline in the configuration. This includes DataSets and parameters."
             )
             switch_catalog_logging(catalog, False)
 
-        add_run_id_to_artifact_datasets(catalog, mlflow.active_run().info.run_id)
-
     @hook_impl
     def before_node_run(
         self, node: Node, catalog: DataCatalog, inputs: Dict[str, Any], is_async: bool