From 37159db633646059eb31efa4654dca1d630d9370 Mon Sep 17 00:00:00 2001 From: Boris Feld Date: Mon, 8 Jul 2024 17:30:51 +0200 Subject: [PATCH] Fix Ray+XGBoost example --- .../ray-train/notebooks/Comet_with_ray_train_xgboost.ipynb | 3 ++- 1 file changed, 2 insertions(+), 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 index 650e433c..7fb9745c 100644 --- 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\"" ] }, {