[Versioning] Add python 3.12 to setup.py (#2282)

.github/unittest/linux/scripts/environment.yml

@@ -28,6 +28,7 @@ dependencies:
     - mlflow
     - av
     - coverage
-    - ray<2.8.0
+    - ray
     - transformers
     - ninja
+    - timm

.github/unittest/linux/scripts/run_all.sh

@@ -88,7 +88,8 @@ conda deactivate
 conda activate "${env_dir}"
 
 echo "installing gymnasium"
-pip3 install "gymnasium[atari,ale-py,accept-rom-license]"
+pip3 install "gymnasium"
+pip3 install ale_py
 pip3 install mo-gymnasium[mujoco]  # requires here bc needs mujoco-py
 pip3 install mujoco -U
 

.github/unittest/linux_distributed/scripts/environment.yml

@@ -27,5 +27,5 @@ dependencies:
     - mlflow
     - av
     - coverage
-    - ray<2.8.0
+    - ray
     - virtualenv

.github/unittest/linux_olddeps/scripts_gym_0_13/environment.yml

@@ -24,5 +24,5 @@ dependencies:
     - dm_control -e git+https://github.com/deepmind/dm_control.git@c053360edea6170acfd9c8f65446703307d9d352#egg={dm_control}
     - patchelf
     - pyopengl==3.1.4
-    - ray<2.8.0
+    - ray
     - av

.github/unittest/linux_optdeps/scripts/environment.yml

@@ -17,4 +17,4 @@ dependencies:
     - pyyaml
     - scipy
     - coverage
-    - ray<2.8.0
+    - ray

.github/workflows/test-linux.yml

@@ -22,7 +22,7 @@ jobs:
   tests-cpu:
     strategy:
       matrix:
-        python_version: ["3.8", "3.9", "3.10", "3.11"]
+        python_version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
@@ -51,7 +51,7 @@ jobs:
   tests-gpu:
     strategy:
       matrix:
-        python_version: ["3.10"]
+        python_version: ["3.11"]
         cuda_arch_version: ["12.1"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main

setup.py

@@ -274,6 +274,7 @@ def _main(argv):
             "Programming Language :: Python :: 3.9",
             "Programming Language :: Python :: 3.10",
             "Programming Language :: Python :: 3.11",
+            "Programming Language :: Python :: 3.12",
             "License :: OSI Approved :: MIT License",
             "Operating System :: OS Independent",
             "Development Status :: 4 - Beta",

test/_utils_internal.py

@@ -56,11 +56,32 @@ def HALFCHEETAH_VERSIONED():
 
 def PONG_VERSIONED():
     # load gym
+    # Gymnasium says that the ale_py behaviour changes from 1.0
+    # but with python 3.12 it is already the case with 0.29.1
+    try:
+        import ale_py  # noqa
+    except ImportError:
+        pass
+
     if gym_backend() is not None:
         _set_gym_environments()
         return _PONG_VERSIONED
 
 
+def BREAKOUT_VERSIONED():
+    # load gym
+    # Gymnasium says that the ale_py behaviour changes from 1.0
+    # but with python 3.12 it is already the case with 0.29.1
+    try:
+        import ale_py  # noqa
+    except ImportError:
+        pass
+
+    if gym_backend() is not None:
+        _set_gym_environments()
+        return _BREAKOUT_VERSIONED
+
+
 def PENDULUM_VERSIONED():
     # load gym
     if gym_backend() is not None:
@@ -69,42 +90,46 @@ def PENDULUM_VERSIONED():
 
 
 def _set_gym_environments():
-    global _CARTPOLE_VERSIONED, _HALFCHEETAH_VERSIONED, _PENDULUM_VERSIONED, _PONG_VERSIONED
+    global _CARTPOLE_VERSIONED, _HALFCHEETAH_VERSIONED, _PENDULUM_VERSIONED, _PONG_VERSIONED, _BREAKOUT_VERSIONED
 
     _CARTPOLE_VERSIONED = None
     _HALFCHEETAH_VERSIONED = None
     _PENDULUM_VERSIONED = None
     _PONG_VERSIONED = None
+    _BREAKOUT_VERSIONED = None
 
 
 @implement_for("gym", None, "0.21.0")
 def _set_gym_environments():  # noqa: F811
-    global _CARTPOLE_VERSIONED, _HALFCHEETAH_VERSIONED, _PENDULUM_VERSIONED, _PONG_VERSIONED
+    global _CARTPOLE_VERSIONED, _HALFCHEETAH_VERSIONED, _PENDULUM_VERSIONED, _PONG_VERSIONED, _BREAKOUT_VERSIONED
 
     _CARTPOLE_VERSIONED = "CartPole-v0"
     _HALFCHEETAH_VERSIONED = "HalfCheetah-v2"
     _PENDULUM_VERSIONED = "Pendulum-v0"
     _PONG_VERSIONED = "Pong-v4"
+    _BREAKOUT_VERSIONED = "Breakout-v4"
 
 
 @implement_for("gym", "0.21.0", None)
 def _set_gym_environments():  # noqa: F811
-    global _CARTPOLE_VERSIONED, _HALFCHEETAH_VERSIONED, _PENDULUM_VERSIONED, _PONG_VERSIONED
+    global _CARTPOLE_VERSIONED, _HALFCHEETAH_VERSIONED, _PENDULUM_VERSIONED, _PONG_VERSIONED, _BREAKOUT_VERSIONED
 
     _CARTPOLE_VERSIONED = "CartPole-v1"
     _HALFCHEETAH_VERSIONED = "HalfCheetah-v4"
     _PENDULUM_VERSIONED = "Pendulum-v1"
     _PONG_VERSIONED = "ALE/Pong-v5"
+    _BREAKOUT_VERSIONED = "ALE/Breakout-v5"
 
 
 @implement_for("gymnasium")
 def _set_gym_environments():  # noqa: F811
-    global _CARTPOLE_VERSIONED, _HALFCHEETAH_VERSIONED, _PENDULUM_VERSIONED, _PONG_VERSIONED
+    global _CARTPOLE_VERSIONED, _HALFCHEETAH_VERSIONED, _PENDULUM_VERSIONED, _PONG_VERSIONED, _BREAKOUT_VERSIONED
 
     _CARTPOLE_VERSIONED = "CartPole-v1"
     _HALFCHEETAH_VERSIONED = "HalfCheetah-v4"
     _PENDULUM_VERSIONED = "Pendulum-v1"
     _PONG_VERSIONED = "ALE/Pong-v5"
+    _BREAKOUT_VERSIONED = "ALE/Breakout-v5"
 
 
 if _has_gym: