[CI] Fix 3.12 gymnasium installation

ghstack-source-id: 35fb3ca1819bcd0d3526a1537053df412a06fe86 Pull Request resolved: #2474
pytorch · Oct 9, 2024 · d5c93fb · d5c93fb
1 parent c8b508e
commit d5c93fb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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?