File size: 663 Bytes
cda4ea5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
[tool.poetry]
name = "cleanba"
version = "0.1.0"
description = ""
authors = ["Costa Huang <[email protected]>"]
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"