Add HF and python 3.8 support to image build

diambra · Feb 18, 2024 · e8dc33b · e8dc33b
1 parent 56d87fa
commit e8dc33b
Show file tree

Hide file tree

Showing 7 changed files with 15 additions and 0 deletions.
diff --git a/images/base/on.arg b/images/base/on.arg
@@ -1,3 +1,4 @@
 3.10-bullseye
 3.9-bullseye
+3.8-bullseye
 3.7-bullseye
diff --git a/images/ray_rllib/Dockerfile b/images/ray_rllib/Dockerfile
@@ -7,10 +7,13 @@ RUN apt-get -qy update && \
     apt-get -qy clean
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8 HOME=/tmp
+ENV HF_HOME=/tmp
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
 RUN --mount=target=/usr/src/arena,type=bind,source=. \
     cp -r /usr/src/arena /tmp/arena && \
     pip install /tmp/arena[ray-rllib] && \
     rm -rf /tmp/arena
+
+RUN pip install huggingface_hub
diff --git a/images/ray_rllib/on.arg b/images/ray_rllib/on.arg
@@ -1,2 +1,3 @@
 3.10-bullseye
 3.9-bullseye
+3.8-bullseye
diff --git a/images/sheeprl/Dockerfile b/images/sheeprl/Dockerfile
@@ -7,10 +7,13 @@ RUN apt-get -qy update && \
     apt-get -qy clean
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
+ENV HF_HOME=/tmp
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
 RUN --mount=target=/usr/src/arena,type=bind,source=. \
     cp -r /usr/src/arena /tmp/arena && \
     pip install /tmp/arena[sheeprl] && \
     rm -rf /tmp/arena
+
+RUN pip install huggingface_hub
diff --git a/images/stable-baselines/Dockerfile b/images/stable-baselines/Dockerfile
@@ -7,6 +7,7 @@ RUN apt-get -qy update && \
     apt-get -qy clean
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
+ENV HF_HOME=/tmp
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
@@ -18,3 +19,5 @@ RUN --mount=target=/usr/src/arena,type=bind,source=. \
     pip install /tmp/arena[stable-baselines] && \
     pip install tensorflow==1.15.0 && \
     rm -rf /tmp/arena
+
+RUN pip install huggingface_hub
diff --git a/images/stable-baselines3/Dockerfile b/images/stable-baselines3/Dockerfile
@@ -7,10 +7,13 @@ RUN apt-get -qy update && \
     apt-get -qy clean
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
+ENV HF_HOME=/tmp
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
 RUN --mount=target=/usr/src/arena,type=bind,source=. \
     cp -r /usr/src/arena /tmp/arena && \
     pip install /tmp/arena[stable-baselines3] && \
     rm -rf /tmp/arena
+
+RUN pip install huggingface_hub
diff --git a/images/stable-baselines3/on.arg b/images/stable-baselines3/on.arg
@@ -1,2 +1,3 @@
 3.10-bullseye
 3.9-bullseye
+3.8-bullseye