|
| 1 | +[project] |
| 2 | +name = "pufferlib" |
| 3 | +version = "3.0.0" |
| 4 | +description = "Fast and sane reinforcement learning" |
| 5 | +requires-python = ">=3.9" |
| 6 | +readme = {file = "README.md", content-type = "text/markdown"} |
| 7 | +license = "MIT" |
| 8 | +license-files = ["LICENSE"] |
| 9 | +authors = [{ name = "Joseph Suarez", email = "jsuarez@puffer.ai" }] |
| 10 | +classifiers=[ |
| 11 | + "Intended Audience :: Science/Research", |
| 12 | + "Intended Audience :: Developers", |
| 13 | + "Programming Language :: Python :: 3.9", |
| 14 | + "Programming Language :: Python :: 3.10", |
| 15 | + "Programming Language :: Python :: 3.11", |
| 16 | + "Programming Language :: Python :: 3.12", |
| 17 | + "Programming Language :: Python :: 3.13", |
| 18 | + "Programming Language :: Python :: 3.14", |
| 19 | +] |
| 20 | +dynamic = ["dependencies"] |
| 21 | + |
| 22 | +# Default Gym/Gymnasium/PettingZoo versions |
| 23 | +# Gym: |
| 24 | +# - 0.26 still has deprecation warnings and is the last version of the package |
| 25 | +# - 0.25 adds a breaking API change to reset, step, and render_modes |
| 26 | +# - 0.24 is broken |
| 27 | +# - 0.22-0.23 triggers deprecation warnings by calling its own functions |
| 28 | +# - 0.21 is the most stable version |
| 29 | +# - <= 0.20 is missing dict methods for gym.spaces.Dict |
| 30 | +# - 0.18-0.21 require setuptools<=65.5.0 |
| 31 | +#'gym==0.23', |
| 32 | +#'gymnasium==0.29.1', |
| 33 | +#'pettingzoo==1.24.1', |
| 34 | +[project.optional-dependencies] |
| 35 | +avalon = [ |
| 36 | + 'gymnasium==0.29.1', |
| 37 | + 'avalon-rl==1.0.0', |
| 38 | +] |
| 39 | + |
| 40 | +atari = [ |
| 41 | + 'gymnasium[accept-rom-license]==0.29.1', |
| 42 | + 'opencv-python==3.4.17.63', |
| 43 | + 'ale_py==0.9.0', |
| 44 | +] |
| 45 | + |
| 46 | +box2d = [ |
| 47 | + 'gymnasium[box2d]==0.29.1', |
| 48 | + 'swig==4.1.1', |
| 49 | +] |
| 50 | + |
| 51 | +bsuite = [ |
| 52 | + 'gym==0.23', |
| 53 | + 'gymnasium==0.29.1', |
| 54 | + 'bsuite==0.3.5', |
| 55 | +] |
| 56 | + |
| 57 | +butterfly = [ |
| 58 | + 'gym==0.23', |
| 59 | + 'gymnasium==0.29.1', |
| 60 | + 'pettingzoo[butterfly]==1.24.1', |
| 61 | +] |
| 62 | + |
| 63 | +classic_control = [ |
| 64 | + 'gym==0.23', |
| 65 | + 'gymnasium==0.29.1', |
| 66 | +] |
| 67 | + |
| 68 | +crafter = [ |
| 69 | + 'gym==0.23', |
| 70 | + 'gymnasium==0.29.1', |
| 71 | + 'crafter==1.8.3', |
| 72 | +] |
| 73 | + |
| 74 | +craftax = [ |
| 75 | + 'gymnasium==0.29.1', |
| 76 | + 'craftax', |
| 77 | +] |
| 78 | + |
| 79 | +dm_control = [ |
| 80 | + 'gym==0.23', |
| 81 | + 'gymnasium==0.29.1', |
| 82 | + 'dm_control==1.0.11', |
| 83 | +] |
| 84 | + |
| 85 | +dm_lab = [ |
| 86 | + 'gym==0.23', |
| 87 | + 'gymnasium==0.29.1', |
| 88 | + 'gym_deepmindlab==0.1.2', |
| 89 | + 'dm_env==1.6', |
| 90 | +] |
| 91 | + |
| 92 | +griddly = [ |
| 93 | + 'gym==0.23', |
| 94 | + 'gymnasium==0.29.1', |
| 95 | + 'griddly==1.6.7', |
| 96 | + 'imageio', |
| 97 | +] |
| 98 | + |
| 99 | +kinetix = [ |
| 100 | + 'gym==0.23', |
| 101 | + 'gymnasium==0.29.1', |
| 102 | + 'kinetix-env', |
| 103 | +] |
| 104 | + |
| 105 | +magent = [ |
| 106 | + 'gym==0.23', |
| 107 | + 'gymnasium==0.29.1', |
| 108 | + 'pettingzoo==1.19.0', |
| 109 | + 'magent==0.2.4', |
| 110 | + # The Magent2 package is broken for now |
| 111 | + #'magent2==0.3.2', |
| 112 | +] |
| 113 | + |
| 114 | +metta = [ |
| 115 | + 'gym==0.23', |
| 116 | + 'gymnasium==0.29.1', |
| 117 | + 'omegaconf', |
| 118 | + 'hydra-core', |
| 119 | + 'duckdb', |
| 120 | + 'raylib>=5.5.0', |
| 121 | + 'metta-common @ git+https://github.com/metta-ai/metta.git@main#subdirectory=common', |
| 122 | + 'metta-mettagrid @ git+https://github.com/metta-ai/metta.git@main#subdirectory=mettagrid', |
| 123 | +] |
| 124 | + |
| 125 | +cogames = [ |
| 126 | + 'gym', |
| 127 | + 'gymnasium', |
| 128 | + 'omegaconf', |
| 129 | + 'hydra-core', |
| 130 | + 'duckdb', |
| 131 | + 'raylib>=5.5.0', |
| 132 | + 'mettagrid @ git+https://github.com/metta-ai/metta.git@main#subdirectory=packages/mettagrid', |
| 133 | + 'cogames @ git+https://github.com/metta-ai/metta.git@main#subdirectory=packages/cogames', |
| 134 | +] |
| 135 | + |
| 136 | +microrts = [ |
| 137 | + 'gym==0.23', |
| 138 | + 'gymnasium==0.29.1', |
| 139 | + 'ffmpeg==1.4', |
| 140 | + 'gym_microrts==0.3.2', |
| 141 | +] |
| 142 | + |
| 143 | +minerl = [ |
| 144 | + 'gym==0.17.0', |
| 145 | + 'gymnasium==0.29.1', |
| 146 | + #'git+https://github.com/minerllabs/minerl' |
| 147 | + # Compatiblity warning with urllib3 and chardet |
| 148 | + #'requests==2.31.0', |
| 149 | +] |
| 150 | + |
| 151 | +minigrid = [ |
| 152 | + 'gym==0.23', |
| 153 | + 'gymnasium==0.29.1', |
| 154 | + 'minigrid==2.3.1', |
| 155 | +] |
| 156 | + |
| 157 | +minihack = [ |
| 158 | + 'gym==0.23', |
| 159 | + 'gymnasium==0.29.1', |
| 160 | + 'minihack==0.1.5', |
| 161 | +] |
| 162 | + |
| 163 | +mujoco = [ |
| 164 | + 'gymnasium[mujoco]==1.0.0', |
| 165 | + 'moviepy', |
| 166 | +] |
| 167 | + |
| 168 | +nethack = [ |
| 169 | + 'gym==0.23', |
| 170 | + 'gymnasium==0.29.1', |
| 171 | + 'nle==0.9.1', |
| 172 | +] |
| 173 | + |
| 174 | +nmmo = [ |
| 175 | + 'gym==0.23', |
| 176 | + 'gymnasium==0.29.1', |
| 177 | + 'pettingzoo==1.24.1', |
| 178 | + 'nmmo>=2.1', |
| 179 | +] |
| 180 | + |
| 181 | +open_spiel = [ |
| 182 | + 'gym==0.23', |
| 183 | + 'gymnasium==0.29.1', |
| 184 | + 'pettingzoo==1.19.0', |
| 185 | + 'open_spiel==1.3', |
| 186 | +] |
| 187 | + |
| 188 | +pokemon_red = [ |
| 189 | + 'gym==0.23', |
| 190 | + 'gymnasium==0.29.1', |
| 191 | + 'pokegym>=0.2.0', |
| 192 | + 'einops==0.6.1', |
| 193 | + 'matplotlib', |
| 194 | + 'scikit-image', |
| 195 | + 'pyboy<2.0.0', |
| 196 | + 'hnswlib==0.7.0', |
| 197 | + 'mediapy', |
| 198 | + 'pandas==2.0.2', |
| 199 | + 'pettingzoo', |
| 200 | + 'websockets', |
| 201 | +] |
| 202 | + |
| 203 | +procgen = [ |
| 204 | + 'gym==0.23', |
| 205 | + 'gymnasium==0.29.1', |
| 206 | + 'stable_baselines3==2.1.0', |
| 207 | + 'procgen-mirror==0.10.7', # Procgen mirror for 3.11 and 3.12 support |
| 208 | + # Note: You need glfw==2.7 after installing for some torch versions |
| 209 | +] |
| 210 | + |
| 211 | +#'smac': [ |
| 212 | +# 'git+https://github.com/oxwhirl/smac.git', |
| 213 | +#], |
| 214 | +#'stable-retro': [ |
| 215 | +# 'git+https://github.com/Farama-Foundation/stable-retro.git', |
| 216 | +#] |
| 217 | + |
| 218 | +slimevolley = [ |
| 219 | + 'gym==0.23', |
| 220 | + 'gymnasium==0.29.1', |
| 221 | + 'slimevolley==0.1.0', |
| 222 | +] |
| 223 | + |
| 224 | +vizdoom = [ |
| 225 | + 'gym==0.23', |
| 226 | + 'gymnasium==0.29.1', |
| 227 | + 'vizdoom==1.2.3', |
| 228 | + 'stable_baselines3==2.1.0', |
| 229 | +] |
| 230 | + |
| 231 | +ray = [ |
| 232 | + 'ray==2.23.0', |
| 233 | +] |
| 234 | + |
| 235 | +tribal-village = [ |
| 236 | + 'gym==0.23', |
| 237 | + 'gymnasium==0.29.1', |
| 238 | + 'numpy<2.0', |
| 239 | + 'tribal-village @ git+https://github.com/Metta-AI/tribal-village.git@main', |
| 240 | +] |
| 241 | + |
| 242 | +cleanrl = [ |
| 243 | + 'stable_baselines3==2.1.0', |
| 244 | + 'tensorboard==2.11.2', |
| 245 | + 'tyro==0.8.6', |
| 246 | +] |
| 247 | + |
| 248 | +[project.scripts] |
| 249 | +puffer = "pufferlib.pufferl:main" |
| 250 | + |
| 251 | +[project.urls] |
| 252 | +Homepage = "https://puffer.ai" |
| 253 | + |
| 254 | +[build-system] |
| 255 | +requires = ["setuptools", "wheel", "Cython", "numpy<2.0", "torch", "pybind11"] |
| 256 | +build-backend = "setuptools.build_meta" |
| 257 | + |
| 258 | +[tool.uv] |
| 259 | +no-build-isolation-package = ["torch"] |
| 260 | + |
0 commit comments