Update Gymnasium to 1.0.0a1 and remove Atari in favour of ale-py new …

…gymnasium implementation

setup.py

@@ -35,7 +35,6 @@ def get_version():
 extras = {
     "gym-v21": ["gym>=0.21.0,<0.26", "pyglet==1.5.11"],
     "gym-v26": ["gym>=0.26.2"],
-    "atari": ["ale-py~=0.8.1"],
     # "imageio" should be "gymnasium[mujoco]>=0.26" but there are install conflicts
     "dm-control": ["dm-control>=1.0.10", "imageio", "h5py>=3.7.0"],
     "dm-control-multi-agent": [
@@ -55,7 +54,6 @@ def get_version():
 extras["testing"] = [
     "pytest==7.1.3",
     "pillow>=9.3.0",
-    "autorom[accept-rom-license]~=0.6.0",
 ]
 
 setup(
@@ -71,7 +69,7 @@ def get_version():
     keywords=["Reinforcement Learning", "game", "RL", "AI"],
     python_requires=">=3.8",
     packages=find_packages(),
-    install_requires=["numpy>=1.18.0", "gymnasium>=0.27.0"],
+    install_requires=["numpy>=1.18.0", "gymnasium==1.0.0a1"],
     tests_require=extras["testing"],
     extras_require=extras,
     classifiers=[
@@ -83,7 +81,4 @@ def get_version():
         "Operating System :: OS Independent",
     ],
     include_package_data=True,
-    entry_points={
-        "gymnasium.envs": ["__root__ = shimmy.registration:register_gymnasium_envs"]
-    },
 )

shimmy/__init__.py

@@ -5,6 +5,10 @@
 
 from shimmy.dm_lab_compatibility import DmLabCompatibilityV0
 from shimmy.openai_gym_compatibility import GymV21CompatibilityV0, GymV26CompatibilityV0
+from shimmy.registration import register_gymnasium_envs
+
+# this registers the environments on `import shimmy`
+register_gymnasium_envs()
 
 
 class NotInstallClass:
@@ -73,7 +77,7 @@ def __call__(self, *args: list[Any], **kwargs: Any):
 ]
 
 
-__version__ = "1.3.0"
+__version__ = "1.4.0"
 
 
 try: