Fix the way files are open in DSS MLflow artifact plugin

Valentin Thorey · Valentin Thorey · commit c6708537bd7f · 2021-12-15T21:30:25.000+01:00
diff --git a/dataikuapi/dss_plugin_mlflow/artifact_repository.py b/dataikuapi/dss_plugin_mlflow/artifact_repository.py
@@ -50,7 +50,7 @@ def log_artifact(self, local_file, artifact_path=None):
         path = (
             os.path.join(self.base_artifact_path, artifact_path) if artifact_path else self.base_artifact_path
         )
-        self.managed_folder.put_file(os.path.join(path, os.path.basename(local_file)), open(local_file))
+        self.managed_folder.put_file(os.path.join(path, os.path.basename(local_file)), open(local_file, "rb"))
 
     def log_artifacts(self, local_dir, artifact_path=None):
         """

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ def log_artifact(self, local_file, artifact_path=None):`
`50`	`50`	`path = (`
`51`	`51`	`os.path.join(self.base_artifact_path, artifact_path) if artifact_path else self.base_artifact_path`
`52`	`52`	`)`
`53`		`- self.managed_folder.put_file(os.path.join(path, os.path.basename(local_file)), open(local_file))`
	`53`	`+ self.managed_folder.put_file(os.path.join(path, os.path.basename(local_file)), open(local_file, "rb"))`
`54`	`54`
`55`	`55`	`def log_artifacts(self, local_dir, artifact_path=None):`
`56`	`56`	`"""`