Fix escaping issue with "mlflow<=VERSION" installation in remote runs (…

…mlflow#387)
ljzzju · Aug 27, 2018 · 43a9e63 · 43a9e63
1 parent b74f58b
commit 43a9e63
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mlflow/projects/databricks.py b/mlflow/projects/databricks.py
@@ -187,7 +187,8 @@ def _run_shell_command_job(project_uri, command, env_vars, cluster_spec):
         },
         # NB: We use <= on the version specifier to allow running projects on pre-release
         # versions, where we will select the most up-to-date mlflow version available.
-        "libraries": [{"pypi": {"package": "mlflow<=%s" % VERSION}}]
+        # Also note, that we escape this so '<' is not treated as a shell pipe.
+        "libraries": [{"pypi": {"package": "'mlflow<=%s'" % VERSION}}]
     }
     run_submit_res = _jobs_runs_submit(req_body_json)
     databricks_run_id = run_submit_res["run_id"]