Use 'scikit-learn' rather than 'sklearn' (#860)

NVIDIA-Merlin · Mar 21, 2023 · 267d01f · 267d01f
1 parent 2b7449b
commit 267d01f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ci/dockerfile.ci b/ci/dockerfile.ci
@@ -16,7 +16,7 @@ RUN pip install fastai fastcore fastprogress fastdownload --no-deps
 
 # install dependencies for systems testing 
 RUN pip install pytest-cov pytest-xdist sphinx-multiversion; pip install -r /nvtabular/requirements/dev.txt
-RUN pip install astroid==2.5.6 'feast<0.20' sklearn
+RUN pip install astroid==2.5.6 'feast<0.20' scikit-learn
 RUN echo 'import sphinx.domains' >> /usr/local/lib/python3.8/dist-packages/sphinx/__init__.py
 RUN HOROVOD_GPU_OPERATIONS=NCCL python -m pip install --no-cache-dir horovod && horovodrun --check-build
 
@@ -29,6 +29,7 @@ RUN pip install tox
 # Update the Merlin repos (to avoid needed to rebuild underlying images to get updates)
 RUN cd /Merlin && git pull origin main
 RUN cd /core/ && git pull origin main && pip install . --no-deps
+RUN cd /dataloader/ && git pull origin main && pip install . --no-deps
 RUN cd /nvtabular/ && git pull origin main && pip install . --no-deps
 RUN cd /systems/ && git pull origin main && pip install . --no-deps
 RUN cd /models/ && git pull origin main && pip install . --no-deps