update branch

LICENSE.txt

@@ -1,4 +1,4 @@
-© 2018-2023, DIAMBRA Inc., https://diambra.ai
+© 2018-2024, DIAMBRA Inc., https://diambra.ai
 
 Redistribution and use in source and binary forms, with or without modification, 
 are permitted provided that the following conditions are met:

README.md

@@ -241,4 +241,4 @@ Paper: <a href="https://arxiv.org/abs/2210.10595" target="_blank">https://arxiv.
 
 DIAMBRA Arena software package is subject to our <a href="https://diambra.ai/terms" target="_blank">Terms of Use</a>. By using it, you accept them in full.
 
-###### DIAMBRA, Inc. © Copyright 2018-2023. All Rights Reserved.
+###### DIAMBRA, Inc. © Copyright 2018-2024. All Rights Reserved.

diambra/arena/engine/interface.py

@@ -5,9 +5,7 @@
 from diambra.engine import Client, model
 import grpc
 
-CONNECTION_FAILED_ERROR_TEXT = """DIAMBRA Arena failed to connect to the Engine Server.
- - If you are running a Python script, are you running with DIAMBRA CLI: `diambra run python script.py`?
- - If you are running a Python Notebook, have you started Jupyter Notebook with DIAMBRA CLI: `diambra run jupyter notebook`?
+CONNECTION_FAILED_ERROR_TEXT = """DIAMBRA Arena failed to connect to the Engine Server. Are you running it with DIAMBRA CLI: `diambra run python script.py`?
 
 See the docs (https://docs.diambra.ai) for additional details, or join DIAMBRA Discord Server (https://diambra.ai/discord) for support."""
 
@@ -32,12 +30,7 @@ def env_init(self, env_settings_pb):
         try:
             response = self.client.EnvInit(env_settings_pb)
         except:
-            try:
-                response = self.client.GetError(model.Empty())
-                exceptionMessage = "Received error message from engine: " + response.errorMessage
-                self.logger.error(exceptionMessage)
-            except:
-                raise Exception(CONNECTION_FAILED_ERROR_TEXT)
+            raise Exception(CONNECTION_FAILED_ERROR_TEXT)
 
         return response
 

images/base/Dockerfile

@@ -8,9 +8,13 @@ RUN apt-get -qy update && \
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
 
+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=. \
     cp -r /usr/src/arena /tmp/arena && \
     pip install /tmp/arena && \
     rm -rf /tmp/arena
+
+ENV HF_HOME=/tmp

images/base/on.arg

@@ -1,3 +1,4 @@
 3.10-bullseye
 3.9-bullseye
+3.8-bullseye
 3.7-bullseye

images/ray_rllib/Dockerfile

@@ -8,9 +8,13 @@ RUN apt-get -qy update && \
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8 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.
 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
+
+ENV HF_HOME=/tmp

images/ray_rllib/on.arg

@@ -1,2 +1,3 @@
 3.10-bullseye
 3.9-bullseye
+3.8-bullseye

images/sheeprl/Dockerfile

@@ -8,9 +8,13 @@ RUN apt-get -qy update && \
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
 
+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=. \
     cp -r /usr/src/arena /tmp/arena && \
     pip install /tmp/arena[sheeprl] && \
     rm -rf /tmp/arena
+
+ENV HF_HOME=/tmp

images/stable-baselines/Dockerfile

@@ -8,6 +8,8 @@ RUN apt-get -qy update && \
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
 
+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
@@ -18,3 +20,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
+
+ENV HF_HOME=/tmp

images/stable-baselines3/Dockerfile

@@ -8,9 +8,13 @@ RUN apt-get -qy update && \
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
 
+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=. \
     cp -r /usr/src/arena /tmp/arena && \
     pip install /tmp/arena[stable-baselines3] && \
     rm -rf /tmp/arena
+
+ENV HF_HOME=/tmp

images/stable-baselines3/on.arg

@@ -1,2 +1,3 @@
 3.10-bullseye
 3.9-bullseye
+3.8-bullseye