From d5c93fb29878fa16bc7a2dd8d815fd74763af0ee Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Wed, 9 Oct 2024 12:02:38 +0100 Subject: [PATCH] [CI] Fix 3.12 gymnasium installation ghstack-source-id: 35fb3ca1819bcd0d3526a1537053df412a06fe86 Pull Request resolved: https://github.com/pytorch/rl/pull/2474 --- .github/unittest/linux/scripts/run_all.sh | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/.github/unittest/linux/scripts/run_all.sh b/.github/unittest/linux/scripts/run_all.sh index 5ba834d35d8..e5edc21fa63 100755 --- a/.github/unittest/linux/scripts/run_all.sh +++ b/.github/unittest/linux/scripts/run_all.sh @@ -88,7 +88,13 @@ conda deactivate conda activate "${env_dir}" echo "installing gymnasium" -pip3 install "gymnasium[atari,accept-rom-license,mujoco]<1.0" mo-gymnasium[mujoco] +if [[ "$PYTHON_VERSION" == "3.12" ]]; then + pip3 install ale-py + pip3 install sympy + pip3 install "gymnasium[accept-rom-license,mujoco]<1.0" mo-gymnasium[mujoco] +else + pip3 install "gymnasium[atari,accept-rom-license,mujoco]<1.0" mo-gymnasium[mujoco] +fi pip3 install "mujoco" -U # sanity check: remove?