pip check

aws · Dec 8, 2024 · 655b042 · 655b042
1 parent 4f1b149
commit 655b042
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/autogluon/training/docker/1.2/py3/Dockerfile.cpu b/autogluon/training/docker/1.2/py3/Dockerfile.cpu
@@ -52,8 +52,8 @@ RUN pip install --no-cache-dir -U --trusted-host pypi.org --trusted-host files.p
  && pip install --no-cache-dir "pillow<11.0.0" \
  && pip install --no-cache-dir "ninja<1.11.1.1" \
  # Fix pip test
- && pip install --no-cache-dir "dill<0.3.9" \
- && pip install --no-cache-dir "multiprocess>0.70.16"
+ && pip install --no-cache-dir "datasets<3.1.0" \
+ && pip install --no-cache-dir "pathos<0.3.3"
 
 
 RUN HOME_DIR=/root \

diff --git a/autogluon/training/docker/1.2/py3/cu124/Dockerfile.gpu b/autogluon/training/docker/1.2/py3/cu124/Dockerfile.gpu
@@ -50,8 +50,8 @@ RUN pip install --no-cache-dir -U --trusted-host pypi.org --trusted-host files.p
  && pip install --no-cache-dir "pillow<11.0.0" \
  && pip install --no-cache-dir "ninja<1.11.1.1" \
   # Fix pip test
- && pip install --no-cache-dir "dill<0.3.9" \
- && pip install --no-cache-dir "multiprocess>0.70.16"
+&& pip install --no-cache-dir "datasets<3.1.0" \
+&& pip install --no-cache-dir "pathos<0.3.3"
 
 
 RUN HOME_DIR=/root \