| [tool.poetry] | |
| name = "cleanrl" | |
| version = "1.0.0" | |
| description = "High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features" | |
| authors = ["Costa Huang <[email protected]>"] | |
| include = ["cleanrl_utils"] | |
| keywords = ["reinforcement", "machine", "learning", "research"] | |
| license="MIT" | |
| readme = "README.md" | |
| [tool.poetry.dependencies] | |
| python = ">=3.7.1,<3.10" | |
| tensorboard = "^2.10.0" | |
| wandb = "^0.13.3" | |
| gym = {version = "0.23.1", extras = ["classic_control"]} | |
| torch = "^1.12.1" | |
| stable-baselines3 = "1.2.0" | |
| [tool.poetry.group.dev.dependencies] | |
| pre-commit = "^2.20.0" | |
| [tool.poetry.group.atari] | |
| optional = true | |
| [tool.poetry.group.atari.dependencies] | |
| ale-py = "0.7.4" | |
| AutoROM = {extras = ["accept-rom-license"], version = "^0.4.2"} | |
| opencv-python = "^4.6.0.66" | |
| [tool.poetry.group.pybullet] | |
| optional = true | |
| [tool.poetry.group.pybullet.dependencies] | |
| pybullet = "3.1.8" | |
| [tool.poetry.group.procgen] | |
| optional = true | |
| [tool.poetry.group.procgen.dependencies] | |
| procgen = "^0.10.7" | |
| [tool.poetry.group.pytest] | |
| optional = true | |
| [tool.poetry.group.pytest.dependencies] | |
| pytest = "^7.1.3" | |
| [tool.poetry.group.mujoco] | |
| optional = true | |
| [tool.poetry.group.mujoco.dependencies] | |
| free-mujoco-py = "^2.1.6" | |
| [tool.poetry.group.docs] | |
| optional = true | |
| [tool.poetry.group.docs.dependencies] | |
| mkdocs-material = "^8.4.3" | |
| markdown-include = "^0.7.0" | |
| [tool.poetry.group.jax] | |
| optional = true | |
| [tool.poetry.group.jax.dependencies] | |
| jax = "^0.3.17" | |
| jaxlib = "^0.3.15" | |
| flax = "^0.6.0" | |
| [tool.poetry.group.optuna] | |
| optional = true | |
| [tool.poetry.group.optuna.dependencies] | |
| optuna = "^3.0.1" | |
| optuna-dashboard = "^0.7.2" | |
| rich = "<12.0" | |
| [tool.poetry.group.envpool] | |
| optional = true | |
| [tool.poetry.group.envpool.dependencies] | |
| envpool = "^0.6.4" | |
| [tool.poetry.group.pettingzoo] | |
| optional = true | |
| [tool.poetry.group.pettingzoo.dependencies] | |
| PettingZoo = "1.18.1" | |
| SuperSuit = "3.4.0" | |
| multi-agent-ale-py = "0.1.11" | |
| [tool.poetry.group.cloud] | |
| optional = true | |
| [tool.poetry.group.cloud.dependencies] | |
| boto3 = "^1.24.70" | |
| awscli = "^1.25.71" | |
| [tool.poetry.group.isaacgym] | |
| optional = true | |
| [tool.poetry.group.isaacgym.dependencies] | |
| isaacgymenvs = {git = "https://github.com/vwxyzjn/IsaacGymEnvs.git", rev = "poetry"} | |
| isaacgym = {path = "cleanrl/ppo_continuous_action_isaacgym/isaacgym", develop = true} | |
| [build-system] | |
| requires = ["poetry-core"] | |
| build-backend = "poetry.core.masonry.api" | |