Skip to content

Commit 6e2db64

Browse files
authored
anchor xgboost to 1.6.2 to make tests pass (#726)
* anchor xgboost to 1.6.2 to make tests pass * adding comment per review
1 parent 37b1ac4 commit 6e2db64

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

docker/dockerfile.merlin

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,8 @@ RUN pip install cupy-cuda117 nvidia-pyindex pybind11 pytest protobuf transformer
8989
RUN pip install betterproto cachetools graphviz nvtx scipy sklearn
9090
RUN pip install tritonclient[all] grpcio-channelz fiddle wandb
9191
RUN pip install git+https://github.com/rapidsai/asvdb.git@main
92-
RUN pip install xgboost lightgbm treelite==2.3.0 treelite_runtime==2.3.0
92+
# xgboost and treelite are pinned from tritonserver support
93+
RUN pip install xgboost==1.6.2 lightgbm treelite==2.3.0 treelite_runtime==2.3.0
9394
RUN pip install lightfm implicit
9495
RUN pip install numba
9596
RUN pip install "cuda-python>=11.5,<12.0"

0 commit comments

Comments
 (0)