From 5388a1d7f0283f7a102efee712959d9e214fda25 Mon Sep 17 00:00:00 2001
From: edknv <109497216+edknv@users.noreply.github.com>
Date: Thu, 1 Dec 2022 10:38:40 -0800
Subject: [PATCH] Replace sklearn by scikit-learn in pip install (#753)

---
 docker/dockerfile.merlin | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin
index b8b167d9d..987d8f20a 100644
--- 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"]
\ No newline at end of file
+ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]