Packaging with pyproject.toml (#180)

setup.py

@@ -1,112 +1,22 @@
 """Setups the project."""
-import itertools
-from typing import Dict, List
 
-from setuptools import find_packages, setup
+import pathlib
+
+from setuptools import setup
 
 
-def get_description():
-    """Gets the description from the readme."""
-    with open("README.md") as file:
-        long_description = ""
-        header_count = 0
-        for line in file:
-            if line.startswith("##"):
-                header_count += 1
-            if header_count < 2:
-                long_description += line
-            else:
-                break
-    return header_count, long_description
+CWD = pathlib.Path(__file__).absolute().parent
 
 
 def get_version():
     """Gets the gymnasium version."""
-    path = "gymnasium/__init__.py"
-    with open(path) as file:
-        lines = file.readlines()
+    path = CWD / "gymnasium" / "__init__.py"
+    content = path.read_text()
 
-    for line in lines:
+    for line in content.splitlines():
         if line.startswith("__version__"):
             return line.strip().split()[-1].strip().strip('"')
     raise RuntimeError("bad version data in __init__.py")
 
 
-# Environment-specific dependencies.
-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"],
-    "mujoco": ["mujoco>=2.3.0", "imageio>=2.14.1"],
-    "toy_text": ["pygame==2.1.0"],
-    "jax": ["jax==0.3.20", "jaxlib==0.3.20"],
-    "other": [
-        "lz4>=3.1.0",
-        "opencv-python>=3.0",
-        "matplotlib>=3.0",
-        "moviepy>=1.0.0",
-        "tensorflow>=2.1.0",
-        "torch>=1.0.0",
-    ],
-}
-
-# All dependency groups - accept rom license as requires user to run
-all_groups = set(extras.keys()) - {"accept-rom-license"}
-extras["all"] = list(
-    set(itertools.chain.from_iterable(map(lambda group: extras[group], all_groups)))
-)
-extras["testing"] = [
-    "pytest==7.1.3",
-]
-
-version = get_version()
-header_count, long_description = get_description()
-
-setup(
-    name="Gymnasium",
-    version=version,
-    author="Farama Foundation",
-    author_email="contact@farama.org",
-    description="A standard API for reinforcement learning and a diverse set of reference environments (formerly Gym)",
-    url="https://gymnasium.farama.org/",
-    license="MIT",
-    license_files=("LICENSE",),
-    long_description=long_description,
-    long_description_content_type="text/markdown",
-    keywords=["Reinforcement Learning", "game", "RL", "AI", "gymnasium"],
-    python_requires=">=3.7",
-    tests_require=extras["testing"],
-    packages=[
-        package for package in find_packages() if package.startswith("gymnasium")
-    ],
-    package_data={
-        "gymnasium": [
-            "envs/mujoco/assets/*.xml",
-            "envs/classic_control/assets/*.png",
-            "envs/toy_text/font/*.ttf",
-            "envs/toy_text/img/*.png",
-            "py.typed",
-        ]
-    },
-    include_package_data=True,
-    install_requires=[
-        "numpy >= 1.21.0",
-        "jax-jumpy >= 0.2.0",
-        "cloudpickle >= 1.2.0",
-        "importlib_metadata >= 4.8.0; python_version < '3.10'",
-        "gymnasium_notices >= 0.0.1",
-        "shimmy>=0.1.0, <1.0",
-    ],
-    classifiers=[
-        "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.7",
-        "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,
-)
+setup(name="gymnasium", version=get_version())