Rearranging instructions in dockerfiles

diambra · Feb 19, 2024 · 4fa011a · 4fa011a
1 parent e8dc33b
commit 4fa011a
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 10 deletions.
diff --git a/images/base/Dockerfile b/images/base/Dockerfile
@@ -7,7 +7,8 @@ RUN apt-get -qy update && \
     apt-get -qy clean
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
-ENV HF_HOME=/tmp
+
+RUN pip install huggingface_hub
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
@@ -16,4 +17,4 @@ RUN --mount=target=/usr/src/arena,type=bind,source=. \
     pip install /tmp/arena && \
     rm -rf /tmp/arena
 
-RUN pip install huggingface_hub
+ENV HF_HOME=/tmp
diff --git a/images/ray_rllib/Dockerfile b/images/ray_rllib/Dockerfile
@@ -7,7 +7,8 @@ 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
+
+RUN pip install huggingface_hub
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
@@ -16,4 +17,4 @@ RUN --mount=target=/usr/src/arena,type=bind,source=. \
     pip install /tmp/arena[ray-rllib] && \
     rm -rf /tmp/arena
 
-RUN pip install huggingface_hub
+ENV HF_HOME=/tmp
diff --git a/images/sheeprl/Dockerfile b/images/sheeprl/Dockerfile
@@ -7,7 +7,8 @@ RUN apt-get -qy update && \
     apt-get -qy clean
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
-ENV HF_HOME=/tmp
+
+RUN pip install huggingface_hub
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
@@ -16,4 +17,4 @@ RUN --mount=target=/usr/src/arena,type=bind,source=. \
     pip install /tmp/arena[sheeprl] && \
     rm -rf /tmp/arena
 
-RUN pip install huggingface_hub
+ENV HF_HOME=/tmp
diff --git a/images/stable-baselines/Dockerfile b/images/stable-baselines/Dockerfile
@@ -7,7 +7,8 @@ RUN apt-get -qy update && \
     apt-get -qy clean
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
-ENV HF_HOME=/tmp
+
+RUN pip install huggingface_hub
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
@@ -20,4 +21,4 @@ RUN --mount=target=/usr/src/arena,type=bind,source=. \
     pip install tensorflow==1.15.0 && \
     rm -rf /tmp/arena
 
-RUN pip install huggingface_hub
+ENV HF_HOME=/tmp
diff --git a/images/stable-baselines3/Dockerfile b/images/stable-baselines3/Dockerfile
@@ -7,7 +7,8 @@ RUN apt-get -qy update && \
     apt-get -qy clean
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
-ENV HF_HOME=/tmp
+
+RUN pip install huggingface_hub
 
 # Copy arena to tmp since bind-mount is read-only and pip doesn't support
 # out-of-tree builds.
@@ -16,4 +17,4 @@ RUN --mount=target=/usr/src/arena,type=bind,source=. \
     pip install /tmp/arena[stable-baselines3] && \
     rm -rf /tmp/arena
 
-RUN pip install huggingface_hub
+ENV HF_HOME=/tmp