Give MLflowHandle the ability to reload artifact_repository_registry

Author: Valentin Thorey

dataikuapi/dss_plugin_mlflow/utils.py

@@ -37,6 +37,14 @@ def __init__(self, client, project_key, managed_folder="mlflow_artifacts", host=
             )
         sys.path.insert(0, self.tempdir)
 
+        # Reload the artifact_repository_registry in case MLflow was imported beforehand
+        import mlflow
+        try:  # python 3.4+
+            from importlib import reload
+        except ImportError:
+            pass
+        reload(mlflow.store.artifact.artifact_repository_registry)
+
         # Setup authentication
         if client._session.auth is not None:
             self.mlflow_env.update({