mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-07-31 22:04:31 +00:00
e913bc81b8
* feat: add `isort` to `pre-commit` * ci: skip `__init__.py` file for `isort` * ci: make `isort` mandatory in lint pipeline * docs: add a section on Git hooks * ci: check isort diff * fix: isort from master branch * docs: add pre-commit badge * ci: update black + bandit versions * feat: add PR template * refactor: PR template * ci: remove bandit * docs: add Black badge * ci: try to remove all `|| true` statements * ci: remove lint_python job - Remove `lint_python` CI job - Move `pyupgrade` job to `pre-commit` workflow * fix: avoid messing with typing * docs: add a note on running `pre-cpmmit` manually * ci: apply `pre-commit` to the whole codebase
74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
import itertools
|
|
import os.path
|
|
import sys
|
|
|
|
from setuptools import find_packages, setup
|
|
|
|
# Don't import gym module here, since deps may not be installed
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "gym"))
|
|
from version import VERSION
|
|
|
|
# Environment-specific dependencies.
|
|
extras = {
|
|
"atari": ["ale-py~=0.7.4"],
|
|
"accept-rom-license": ["autorom[accept-rom-license]~=0.4.2"],
|
|
"box2d": ["box2d-py==2.3.5", "pygame==2.1.0"],
|
|
"classic_control": ["pygame==2.1.0"],
|
|
"mujoco": ["mujoco_py>=1.50, <2.0"],
|
|
"toy_text": ["pygame==2.1.0", "scipy>=1.4.1"],
|
|
"other": ["lz4>=3.1.0", "opencv-python>=3.0"],
|
|
}
|
|
|
|
# Meta dependency groups.
|
|
nomujoco_blacklist = {"mujoco", "accept-rom-license", "atari"}
|
|
nomujoco_groups = set(extras.keys()) - nomujoco_blacklist
|
|
|
|
extras["nomujoco"] = list(
|
|
itertools.chain.from_iterable(map(lambda group: extras[group], nomujoco_groups))
|
|
)
|
|
|
|
|
|
all_blacklist = {"accept-rom-license"}
|
|
all_groups = set(extras.keys()) - all_blacklist
|
|
|
|
extras["all"] = list(
|
|
itertools.chain.from_iterable(map(lambda group: extras[group], all_groups))
|
|
)
|
|
|
|
setup(
|
|
name="gym",
|
|
version=VERSION,
|
|
description="Gym: A universal API for reinforcement learning environments",
|
|
url="https://www.gymlibrary.ml/",
|
|
author="Gym Community",
|
|
author_email="jkterry@umd.edu",
|
|
license="MIT",
|
|
packages=[package for package in find_packages() if package.startswith("gym")],
|
|
zip_safe=False,
|
|
install_requires=[
|
|
"numpy>=1.18.0",
|
|
"cloudpickle>=1.2.0",
|
|
"importlib_metadata>=4.10.0; python_version < '3.10'",
|
|
"gym_notices>=0.0.4",
|
|
],
|
|
extras_require=extras,
|
|
package_data={
|
|
"gym": [
|
|
"envs/mujoco/assets/*.xml",
|
|
"envs/classic_control/assets/*.png",
|
|
"envs/toy_text/font/*.ttf",
|
|
"envs/toy_text/img/*.png",
|
|
"py.typed",
|
|
]
|
|
},
|
|
tests_require=["pytest", "mock"],
|
|
python_requires=">=3.7",
|
|
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",
|
|
],
|
|
)
|