Skip to content

Commit 04e4bcb

Browse files
committed
adding optuna and plotly to dockerfile
1 parent e98bbbc commit 04e4bcb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docker/dockerfile.merlin

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ RUN pip install "cmake<3.25.0" ninja scikit-build
9595
RUN pip install pandas==1.3.5
9696
RUN pip install cupy-cuda117 nvidia-pyindex pybind11 pytest protobuf transformers==4.12 tensorflow-metadata
9797
RUN pip install betterproto cachetools graphviz nvtx scipy sklearn
98-
RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array
98+
RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array optuna plotly
9999
RUN pip install git+https://github.com/rapidsai/asvdb.git@main
100100
# xgboost and treelite are pinned from tritonserver support
101101
RUN pip install xgboost==1.6.2 lightgbm treelite==2.3.0 treelite_runtime==2.3.0

0 commit comments

Comments
 (0)