get_or_create_conda_env now returns an environment object

ncsa · Sep 5, 2023 · 6b1a1d9 · 6b1a1d9
1 parent ec8cece
commit 6b1a1d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlflow_slurm/slurm_backend.py b/mlflow_slurm/slurm_backend.py
@@ -187,8 +187,8 @@ def run(self, project_uri: str, entry_point: str, params: dict, version: str,
             tracking.MlflowClient().set_tag(active_run.info.run_id, MLFLOW_PROJECT_ENV,
                                             "conda")
             command_separator = " && "
-            conda_env_name = get_or_create_conda_env(project.env_config_path)
-            command_args += get_conda_command(conda_env_name)
+            conda_env = get_or_create_conda_env(project.env_config_path)
+            command_args += conda_env.get_activate_command()
         else:
             _logger.fatal(f"Unknown project environment type provided: {project.env_type}")
             return None