Fix: support models saved with mlflow=2 in mlflowserver conda_env_cre…

…ate (#4505)
SeldonIO · Jan 10, 2023 · 3590848 · 3590848
1 parent 6179015
commit 3590848
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/servers/mlflowserver/mlflowserver/conda_env_create.py b/servers/mlflowserver/mlflowserver/conda_env_create.py
@@ -48,6 +48,8 @@ def setup_env(model_folder):
     flavours = mlmodel["flavors"]
     pyfunc_flavour = flavours["python_function"]
     env_file_name = pyfunc_flavour["env"]
+    if isinstance(env_file_name, dict):
+        env_file_name = env_file_name["conda"]
     env_file_path = os.path.join(model_folder, env_file_name)
     env_file_path = copy_env(env_file_path)