Skip to content

Commit 2e26e09

Browse files
committed
Correct MLflow model archive filename
1 parent 00696df commit 2e26e09

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dataikuapi/dss/savedmodel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def import_mlflow_version_from_path(self, version_id, path, code_env_name = "INH
140140
try:
141141
archive_filename = shutil.make_archive(os.path.join(archive_temp_dir, "tmpmodel"), "zip", path) #[, root_dir[, base_dir[, verbose[, dry_run[, owner[, group[, logger]]]]]]])
142142

143-
with open("tmpmodel.zip", "rb") as fp:
143+
with open(archive_filename, "rb") as fp:
144144
self.client._perform_empty("POST", "/projects/%s/savedmodels/%s/versions/%s?codeEnvName=%s" % (self.project_key, self.sm_id, version_id, code_env_name),
145145
files={"file":(archive_filename, fp)})
146146
return self.get_mlflow_version_handler(version_id)

0 commit comments

Comments
 (0)