fix modern dataset

Galileo-Galilei · Oct 15, 2024 · b2138c1 · b2138c1
1 parent 65f8ae1
commit b2138c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kedro_mlflow/io/artifacts/mlflow_artifact_dataset.py b/kedro_mlflow/io/artifacts/mlflow_artifact_dataset.py
@@ -62,7 +62,7 @@ def _save(self, data: Any):
                 # for logging on remote storage like Azure S3
                 local_path = local_path.as_posix()
 
-                if getattr(super().save, "__savewrapped__", False):  # modern dataset
+                if getattr(super().save, "__wrapped__", False):  # modern dataset
                     super().save.__wrapped__(self, data)
                 else:  # legacy dataset
                     super()._save(data)
@@ -134,7 +134,7 @@ def _load(self) -> Any:  # pragma: no cover
                     shutil.copy(src=temp_download_filepath, dst=local_path)
 
                 # finally, read locally
-                if getattr(super().load, "__loadwrapped__", False):  # modern dataset
+                if getattr(super().load, "__wrapped__", False):  # modern dataset
                     return super().load.__wrapped__(self)
                 else:  # legacy dataset
                     return super()._load()