From 04e4bcb0e02b88cc3d2ea7959359c38e9b8dcd9b Mon Sep 17 00:00:00 2001 From: Julio Date: Sun, 20 Nov 2022 18:42:54 -0500 Subject: [PATCH] adding optuna and plotly to dockerfile --- docker/dockerfile.merlin | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin index 577a7ffed..b844e64b6 100644 --- a/docker/dockerfile.merlin +++ b/docker/dockerfile.merlin @@ -95,7 +95,7 @@ RUN pip install "cmake<3.25.0" ninja scikit-build RUN pip install pandas==1.3.5 RUN pip install cupy-cuda117 nvidia-pyindex pybind11 pytest protobuf transformers==4.12 tensorflow-metadata RUN pip install betterproto cachetools graphviz nvtx scipy sklearn -RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array +RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array optuna plotly RUN pip install git+https://github.com/rapidsai/asvdb.git@main # xgboost and treelite are pinned from tritonserver support RUN pip install xgboost==1.6.2 lightgbm treelite==2.3.0 treelite_runtime==2.3.0