[tune] Use latest Ax release (#11155)

* Use latest Ax release * specify egg * Install latest Ax only for py3.7
2026-07-03 20:39:52 +08:00 · 2020-10-01 16:51:15 +01:00
parent 36bda8432b
commit de3c606b20
1 changed files with 5 additions and 0 deletions
@@ -291,6 +291,11 @@ install_dependencies() {
  # Additional Tune test dependencies.
  if [ "${TUNE_TESTING-}" = 1 ]; then
    pip install -r "${WORKSPACE_DIR}"/python/requirements_tune.txt
+
+    # Temporary fix to support latest Ax release on Python 3.7
+    if [ "${PYTHON}" = "3.7" ]; then
+      pip install -U git+https://github.com/facebook/Ax.git@0.1.16#egg=ax-platform
+    fi
  fi

  # Additional Tune dependency for Horovod.