🐛 Force return on load for old style datasets (#592)

Galileo-Galilei · Sep 23, 2024 · 8ea5a57 · 8ea5a57
1 parent ed43e0d
commit 8ea5a57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kedro_mlflow/io/artifacts/mlflow_artifact_dataset.py b/kedro_mlflow/io/artifacts/mlflow_artifact_dataset.py
@@ -137,7 +137,7 @@ def _load(self) -> Any:  # pragma: no cover
                 if getattr(super().load, "__loadwrapped__", False):  # modern dataset
                     return super().load.__wrapped__(self)
                 else:  # legacy dataset
-                    super()._load()
+                    return super()._load()
 
         # rename the class
         parent_name = dataset_obj.__name__