|
1 | 1 | from setuptools import find_packages, setup |
2 | 2 |
|
3 | | -GYM_VERSION = "0.29.1" |
4 | | -PETTINGZOO_VERSION = "1.24.2" |
| 3 | +GYMNASIUM_VERSION = "0.29.1" |
| 4 | +PETTINGZOO_VERSION = "1.24.3" |
5 | 5 |
|
6 | 6 |
|
7 | 7 | extras = { |
8 | 8 | "atari": [ |
9 | | - "gymnasium[atari, accept-rom-license]~={}".format(GYM_VERSION), |
| 9 | + f"gymnasium[atari, accept-rom-license]~={GYMNASIUM_VERSION}", |
10 | 10 | ], |
11 | 11 | "box2d": [ |
12 | | - "gymnasium[box2d]~={}".format(GYM_VERSION), |
| 12 | + f"gymnasium[box2d]~={GYMNASIUM_VERSION}", |
13 | 13 | ], |
14 | 14 | "pybullet": [ |
15 | | - "pybullet>=3.2.2", |
| 15 | + "pybullet>=3.2.2,<4", |
16 | 16 | "gym>=0.10.0,<0.26.0", |
17 | 17 | ], |
18 | 18 | "mujoco": [ |
19 | | - "gymnasium[mujoco]~={}".format(GYM_VERSION), |
| 19 | + f"gymnasium[mujoco]~={GYMNASIUM_VERSION}", |
20 | 20 | ], |
21 | 21 | "ma-atari": [ |
22 | | - "PettingZoo[atari, accept-rom-license]~={}".format(PETTINGZOO_VERSION), |
23 | | - "supersuit~=3.9.1", |
| 22 | + f"PettingZoo[atari, accept-rom-license]~={PETTINGZOO_VERSION}", |
| 23 | + "supersuit~=3.9.2", |
24 | 24 | ], |
25 | 25 | "test": [ |
26 | | - "black>=24.1.1", # linting/formatting |
27 | | - "isort>=5.13.2", # sort imports |
28 | | - "flake8>=7.0.0", # more linting |
29 | | - "torch-testing>=0.0.2", # pytorch assertion library |
| 26 | + "black~=24.2.0", # linting/formatting |
| 27 | + "isort~=5.13.2", # sort imports |
| 28 | + "flake8~=7.0.0", # more linting |
| 29 | + "torch-testing==0.0.2", # pytorch assertion library |
30 | 30 | ], |
31 | 31 | "docs": [ |
32 | | - "sphinx>=3.2.1", # documentation library |
33 | | - "sphinx-autobuild>=2020.9.1", # documentation live reload |
34 | | - "sphinx-rtd-theme>=0.5.0", # documentation theme |
35 | | - "sphinx-automodapi>=0.13", # autogenerate docs for modules |
| 32 | + "sphinx~=3.2.1", # documentation library |
| 33 | + "sphinx-autobuild~=2020.9.1", # documentation live reload |
| 34 | + "sphinx-rtd-theme~=0.5.0", # documentation theme |
| 35 | + "sphinx-automodapi~=0.13.0", # autogenerate docs for modules |
36 | 36 | ], |
37 | 37 | } |
38 | 38 |
|
|
71 | 71 | ], |
72 | 72 | }, |
73 | 73 | install_requires=[ |
74 | | - "gymnasium~={}".format(GYM_VERSION), # common environment interface |
75 | | - "numpy>=1.22.3", # math library |
76 | | - "matplotlib>=3.5.1", # plotting library |
77 | | - "opencv-python-headless>=4.0.0", # used by atari wrappers |
78 | | - "torch>=2.0.0", # core deep learning library |
79 | | - "tensorboard>=2.8.0", # logging and visualization |
80 | | - "cloudpickle>=2.0.0", # used to copy environments |
| 74 | + f"gymnasium~={GYMNASIUM_VERSION}", # common environment interface |
| 75 | + "numpy~=1.22", # math library |
| 76 | + "matplotlib~=3.7", # plotting library |
| 77 | + "opencv-python-headless~=4.0", # used by atari wrappers |
| 78 | + "torch~=2.0", # core deep learning library |
| 79 | + "tensorboard~=2.8", # logging and visualization |
| 80 | + "cloudpickle~=2.0", # used to copy environments |
81 | 81 | ], |
82 | 82 | extras_require=extras, |
83 | 83 | ) |
0 commit comments