ax-platform didn't fix the pandas issue

Signed-off-by: Ben Wilson <benjamin.wilson@databricks.com>
mlflow · Mar 15, 2023 · 7094c84 · 7094c84
1 parent 1201afe
commit 7094c84
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/examples/pytorch/AxHyperOptimizationPTL/conda.yaml b/examples/pytorch/AxHyperOptimizationPTL/conda.yaml
@@ -17,5 +17,7 @@ dependencies:
       # required by many versions of pytorch
       - gpytorch<1.9.0
       - protobuf<4.0.0
+      # Pinning pandas version less than 1.4.4 due to https://github.com/facebook/Ax/issues/1153
+      - pandas<=1.4.4
       # TODO: Remove this requirement once ax-platform achieves compatibility with SQLAlchemy 2.x
       - sqlalchemy<2
diff --git a/examples/pytorch/AxHyperOptimizationPTL/python_env.yaml b/examples/pytorch/AxHyperOptimizationPTL/python_env.yaml
@@ -15,5 +15,7 @@ dependencies:
   # required by many versions of pytorch
   - gpytorch<1.9.0
   - protobuf<4.0.0
+  # Pinning pandas version less than 1.4.4 due to https://github.com/facebook/Ax/issues/1153
+  - pandas<=1.4.4
   # TODO: Remove this requirement once ax-platform achieves compatibility with SQLAlchemy 2.x
   - sqlalchemy<2