Fix test

mars-project · Oct 29, 2023 · 314b628 · 314b628
1 parent 6c97cc6
commit 314b628
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/.github/workflows/platform-ci.yml b/.github/workflows/platform-ci.yml
@@ -106,13 +106,10 @@ jobs:
               rm -fr /tmp/etcd-$ETCD_VER-linux-amd64.tar.gz /tmp/etcd-download-test
             fi
             if [ -n "$WITH_RAY" ] || [ -n "$WITH_RAY_DAG" ] || [ -n "$WITH_RAY_DEPLOY" ]; then
-              pip install "ray>=1.8.0,<2.4.0"
-              pip install xgboost_ray "protobuf<4"
+              pip install "ray>=1.8.0,<2.4.0" "xgboost<2" "xgboost_ray<0.1.14" "protobuf<4"
               # Ray Datasets need pyarrow>=6.0.1
               pip install "pyarrow>=6.0.1"
               pip install lightgbm
-              # patch xgboost_ray and xgboost incompatibility
-              cp ./ci/xgboost_ray_patch.pth "$(python -c 'import site; print(site.getsitepackages()[0])')"
             fi
             if [ -n "$RUN_DASK" ]; then
               pip install "dask[complete]" "mimesis<9.0.0" scikit-learn

diff --git a/ci/xgboost_ray_patch.pth b/ci/xgboost_ray_patch.pth