| [tool.poetry] |
| name = "cleanba" |
| version = "0.1.0" |
| description = "" |
| authors = ["Costa Huang <costa.huang@outlook.com>"] |
| readme = "README.md" |
| packages = [ |
| { include = "cleanba" }, |
| { include = "cleanrl_utils" }, |
| ] |
|
|
| [tool.poetry.dependencies] |
| python = "^3.8" |
| tensorboard = "^2.12.0" |
| envpool = "^0.8.1" |
| jax = "0.3.25" |
| flax = "0.6.0" |
| optax = "0.1.3" |
| huggingface-hub = "^0.12.0" |
| jaxlib = "0.3.25" |
| wandb = "^0.13.10" |
| tensorboardx = "^2.5.1" |
| chex = "0.1.5" |
| gym = "0.23.1" |
| opencv-python = "^4.7.0.68" |
| moviepy = "^1.0.3" |
|
|
|
|
| [tool.poetry.group.dev.dependencies] |
| pre-commit = "^3.0.4" |
|
|
| [build-system] |
| requires = ["poetry-core"] |
| build-backend = "poetry.core.masonry.api" |
|
|
