scikit (#538)

montanalow · web-flow · commit 260f8739dfdc · 2023-02-08T09:52:05.000-08:00
Co-authored-by: Montana Low &lt;montana.low@gmail.com&gt;
diff --git a/pgml-extension/Dockerfile.local b/pgml-extension/Dockerfile.local
@@ -11,7 +11,7 @@ RUN cat /etc/apt/sources.list
 RUN apt-get update && apt-get install -y postgresql-pgml-14
 
 # Cache this, quicker
-RUN pip3 install xgboost sklearn diptest torch lightgbm transformers datasets sentencepiece sacremoses sacrebleu rouge
+RUN pip3 install xgboost skikit-learn diptest torch lightgbm transformers datasets sentencepiece sacremoses sacrebleu rouge
 
 COPY --chown=postgres:postgres . /app
 WORKDIR /app
