Fix Ray+XGBoost example

comet-ml · Jul 8, 2024 · 37159db · 37159db
1 parent 8149415
commit 37159db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/integrations/model-training/ray-train/notebooks/Comet_with_ray_train_xgboost.ipynb b/integrations/model-training/ray-train/notebooks/Comet_with_ray_train_xgboost.ipynb
@@ -41,7 +41,8 @@
    },
    "outputs": [],
    "source": [
-    "%pip install -U comet_ml \"ray[air]>=2.1.0\" xgboost_ray \"pandas!=2.2.0\""
+    "# XGBoost is pinned because of https://github.com/ray-project/ray/issues/46476\n",
+    "%pip install -U comet_ml \"ray[air]>=2.1.0\" xgboost_ray \"pandas!=2.2.0\" \"xgboost!=2.1.0\""
    ]
   },
   {