From e4a8473ea938c445f75d6d2ed08e7edf737297dd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ludovic=20Pe=CC=81net?= Date: Mon, 27 Sep 2021 15:02:53 +0200 Subject: [PATCH] Add code_env_name parameter to import_mlflow_version_from_path, to match backend evolution. --- dataikuapi/dss/savedmodel.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/dataikuapi/dss/savedmodel.py b/dataikuapi/dss/savedmodel.py index 84a0ba43..67b440e9 100644 --- a/dataikuapi/dss/savedmodel.py +++ b/dataikuapi/dss/savedmodel.py @@ -117,7 +117,7 @@ def get_origin_ml_task(self): if fmi is not None: return DSSMLTask.from_full_model_id(self.client, fmi, project_key=self.project_key) - def import_mlflow_version_from_path(self, version_id, path): + def import_mlflow_version_from_path(self, version_id, path, code_env_name="INHERIT"): """ Create a new version for this saved model from a path containing a MLFlow model. @@ -125,6 +125,7 @@ def import_mlflow_version_from_path(self, version_id, path): :param str version_id: Identifier of the version to create :param str path: An absolute path on the local filesystem. Must be a folder, and must contain a MLFlow model + :param str code_env_name: Name of the code environment to use for this model. Default is to inherit from project configuration :return a :class:MLFlowVersionHandler in order to interact with the new MLFlow model version """ @@ -136,7 +137,7 @@ def import_mlflow_version_from_path(self, version_id, path): with open("tmpmodel.zip", "rb") as fp: self.client._perform_empty("POST", "/projects/%s/savedmodels/%s/versions/%s" % (self.project_key, self.sm_id, version_id), - files={"file":("tmpmodel.zip", fp)}) + files={"file":("tmpmodel.zip", fp)}, params={"codeEnvName": code_env_name}) return self.get_mlflow_version_handler(version_id)