From 3dfb4a396d51c40c3c496b756819d646e0cde857 Mon Sep 17 00:00:00 2001
From: Mark Towers <mark.m.towers@gmail.com>
Date: Thu, 1 Dec 2022 12:51:17 +0000
Subject: [PATCH] Add support for python 3.11 (#71)

---
 .github/workflows/build-publish.yml |  3 +++
 README.md                           |  2 +-
 requirements.txt                    |  2 +-
 setup.py                            | 11 ++++++-----
 test_requirements.txt               |  2 +-
 5 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/build-publish.yml b/.github/workflows/build-publish.yml
index e8bc0582c..701478daf 100644
--- a/.github/workflows/build-publish.yml
+++ b/.github/workflows/build-publish.yml
@@ -32,6 +32,9 @@ jobs:
         - os: ubuntu-latest
           python: 310
           platform: manylinux_x86_64
+        - os: ubuntu-latest
+          python: 311
+          platform: manylinux_x86_64
 
     steps:
     - uses: actions/checkout@v3
diff --git a/README.md b/README.md
index fcfb991f6..797a5e383 100644
--- a/README.md
+++ b/README.md
@@ -27,7 +27,7 @@ To install the base Gymnasium library, use `pip install gymnasium`
 
 This does not include dependencies for all families of environments (there's a massive number, and some can be problematic to install on certain systems). You can install these dependencies for one family like `pip install "gymnasium[atari]"` or use `pip install "gymnasium[all]"` to install all dependencies.
 
-We support and test for Python 3.7, 3.8, 3.9 and 3.10 on Linux and macOS. We will accept PRs related to Windows, but do not officially support it.
+We support and test for Python 3.7, 3.8, 3.9, 3.10, 3.11 on Linux and macOS. We will accept PRs related to Windows, but do not officially support it.
 
 ## API
 
diff --git a/requirements.txt b/requirements.txt
index 84fce9cff..080d7c0fd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,7 +8,7 @@ opencv-python>=3.0
 lz4>=3.1.0
 matplotlib>=3.0
 box2d-py==2.3.5
-pygame==2.1.0
+pygame==2.1.3.dev8
 ale-py~=0.8.0
 mujoco==2.2
 mujoco_py<2.2,>=2.1
diff --git a/setup.py b/setup.py
index 575e5298e..885e084d0 100644
--- a/setup.py
+++ b/setup.py
@@ -36,12 +36,12 @@ def get_version():
 extras: Dict[str, List[str]] = {
     "atari": ["shimmy[atari]>=0.1.0,<1.0"],
     "accept-rom-license": ["autorom[accept-rom-license]~=0.4.2"],
-    "box2d": ["box2d-py==2.3.5", "pygame==2.1.0", "swig==4.*"],
-    "classic_control": ["pygame==2.1.0"],
-    "mujoco_py": ["mujoco_py<2.2,>=2.1"],
+    "box2d": ["box2d-py==2.3.5", "pygame==2.1.3.dev8", "swig==4.*"],
+    "classic_control": ["pygame==2.1.3.dev8"],
+    "mujoco_py": ["mujoco_py"],
     "mujoco": ["mujoco==2.2", "imageio>=2.14.1"],
-    "toy_text": ["pygame==2.1.0"],
-    "jax": ["jax==0.3.20", "jaxlib==0.3.20"],
+    "toy_text": ["pygame==2.1.3.dev8"],
+    "jax": ["jax==0.3.25", "jaxlib==0.3.25"],
     "other": ["lz4>=3.1.0", "opencv-python>=3.0", "matplotlib>=3.0", "moviepy>=1.0.0"],
 }
 
@@ -97,6 +97,7 @@ setup(
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
     ],
     extras_require=extras,
     zip_safe=False,
diff --git a/test_requirements.txt b/test_requirements.txt
index bfe601354..541c2dac4 100644
--- a/test_requirements.txt
+++ b/test_requirements.txt
@@ -4,7 +4,7 @@ opencv-python>=3.0
 mujoco==2.2.0
 matplotlib>=3.0
 imageio>=2.14.1
-pygame==2.1.0
+pygame==2.1.3.dev8
 mujoco_py<2.2,>=2.1
 pytest==7.1.3
 gym==0.26.2