Replace sklearn by scikit-learn in pip install (#753)

NVIDIA-Merlin · Dec 1, 2022 · 5388a1d · 5388a1d
1 parent c2f9519
commit 5388a1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin
@@ -92,7 +92,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 betterproto cachetools graphviz nvtx scipy scikit-learn
 RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array
 RUN pip install git+https://github.com/rapidsai/asvdb.git@main
 # xgboost and treelite are pinned from tritonserver support
@@ -359,4 +359,4 @@ RUN git clone https://github.com/NVIDIA-Merlin/Transformers4Rec.git /transformer
 
 HEALTHCHECK NONE
 CMD ["/bin/bash"]
-ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
+ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]