diff --git a/images/base/Dockerfile b/images/base/Dockerfile index 70415c2..33498fb 100644 --- a/images/base/Dockerfile +++ b/images/base/Dockerfile @@ -10,6 +10,8 @@ ENV LANG=C.UTF-8 LC_ALL=C.UTF-8 RUN pip install huggingface_hub +RUN pip install huggingface_hub + # 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=. \ diff --git a/images/ray_rllib/Dockerfile b/images/ray_rllib/Dockerfile index cc3e66f..ea5bb4a 100644 --- a/images/ray_rllib/Dockerfile +++ b/images/ray_rllib/Dockerfile @@ -10,6 +10,8 @@ ENV LANG=C.UTF-8 LC_ALL=C.UTF-8 HOME=/tmp RUN pip install huggingface_hub +RUN pip install huggingface_hub + # 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=. \ diff --git a/images/stable-baselines/Dockerfile b/images/stable-baselines/Dockerfile index 09fe03a..0cb0291 100644 --- a/images/stable-baselines/Dockerfile +++ b/images/stable-baselines/Dockerfile @@ -10,6 +10,8 @@ ENV LANG=C.UTF-8 LC_ALL=C.UTF-8 RUN pip install huggingface_hub +RUN pip install huggingface_hub + # Copy arena to tmp since bind-mount is read-only and pip doesn't support # out-of-tree builds. # Need to pin the specific version of tensorflow 1.15.0 due to diff --git a/images/stable-baselines3/Dockerfile b/images/stable-baselines3/Dockerfile index 936625d..ac2e0e5 100644 --- a/images/stable-baselines3/Dockerfile +++ b/images/stable-baselines3/Dockerfile @@ -10,6 +10,8 @@ ENV LANG=C.UTF-8 LC_ALL=C.UTF-8 RUN pip install huggingface_hub +RUN pip install huggingface_hub + # 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=. \