Release v2.5.0 (#483)

* Update dockerfile

* Release v2.5.0

* [ci skip] Fix Dockerfile

Author: araffin

CHANGELOG.md

@@ -1,4 +1,4 @@
-## Release 2.5.0a1 (WIP)
+## Release 2.5.0 (2025-01-27)
 
 ### Breaking Changes
 - Upgraded to Pytorch >= 2.3.0
@@ -13,7 +13,7 @@
 ### Documentation
 
 ### Other
-
+- Updated Dockerfile
 
 ## Release 2.4.0 (2024-11-18)
 

docker/Dockerfile

@@ -1,5 +1,4 @@
 ARG PARENT_IMAGE
-ARG USE_GPU
 FROM $PARENT_IMAGE
 
 USER root
@@ -15,16 +14,23 @@ RUN apt-get -y update \
 
 USER $MAMBA_USER
 
-ENV CODE_DIR /home/$MAMBA_USER/code
+ENV CODE_DIR=/home/$MAMBA_USER/code
 COPY requirements.txt /tmp/
 
+# Copy setup file only to install dependencies
+COPY --chown=$MAMBA_USER:$MAMBA_USER ./setup.py ${CODE_DIR}/rl_zoo3/setup.py
+COPY --chown=$MAMBA_USER:$MAMBA_USER ./rl_zoo3/version.txt ${CODE_DIR}/rl_zoo3/rl_zoo3/version.txt
+COPY --chown=$MAMBA_USER:$MAMBA_USER ./hyperparams ${CODE_DIR}/rl_zoo3/hyperparams
+COPY --chown=$MAMBA_USER:$MAMBA_USER ./rl_zoo3/plots ${CODE_DIR}/rl_zoo3/rl_zoo3/plots
+
+
 RUN \
     eval "$(micromamba shell hook --shell bash)" && \
     micromamba activate && \
-    mkdir -p ${CODE_DIR}/rl_zoo3 && \
-    pip uninstall -y stable-baselines3 && \
-    pip install -r /tmp/requirements.txt && \
-    pip install highway-env>=1.7.1 && \
-    pip cache purge
+    cd ${CODE_DIR}/rl_zoo3 && \
+    uv pip uninstall stable-baselines3 && \
+    uv pip install --system -r /tmp/requirements.txt && \
+    uv pip install --system -e ".[plots,tests]" && \
+    uv cache clean
 
 CMD /bin/bash

requirements.txt

@@ -1,5 +1,5 @@
 gym==0.26.2
-stable-baselines3[extra,tests,docs]>=2.5.0a0,<3.0
+stable-baselines3[extra,tests,docs]>=2.5.0,<3.0
 box2d-py==2.3.8
 pybullet_envs_gymnasium>=0.5.0
 # minigrid

rl_zoo3/version.txt

@@ -1 +1 @@
-2.5.0a1
+2.5.0

scripts/build_docker.sh

@@ -3,7 +3,7 @@
 PARENT=stablebaselines/stable-baselines3
 
 TAG=stablebaselines/rl-baselines3-zoo
-VERSION=2.2.0a1
+VERSION=$(cat ./rl_zoo3/version.txt)
 
 if [[ ${USE_GPU} == "True" ]]; then
   PARENT="${PARENT}:${VERSION}"
@@ -12,7 +12,7 @@ else
   TAG="${TAG}-cpu"
 fi
 
-docker build --build-arg PARENT_IMAGE=${PARENT} --build-arg USE_GPU=${USE_GPU} -t ${TAG}:${VERSION} . -f docker/Dockerfile
+docker build --build-arg PARENT_IMAGE=${PARENT} -t ${TAG}:${VERSION} . -f docker/Dockerfile
 docker tag ${TAG}:${VERSION} ${TAG}:latest
 
 if [[ ${RELEASE} == "True" ]]; then

setup.py

@@ -15,7 +15,7 @@
 See https://github.com/DLR-RM/rl-baselines3-zoo
 """
 install_requires = [
-    "sb3_contrib>=2.5.0a0,<3.0",
+    "sb3_contrib>=2.5.0,<3.0",
     "gymnasium>=0.29.1,<1.1.0",
     "huggingface_sb3>=3.0,<4.0",
     "tqdm",