[tool.poetry]
name = "rl"
version = "0.1.0"
description = ""
authors = ["jackson <1666825283@qq.com>"]
readme = "README.md"
packages = [
    {include = "rl"},
    {include = "rl_utils"},
]


[tool.poetry.dependencies]
python = "^3.10"
gymnasium = {extras = ["box2d"], version = "^1.0.0"}
tensorboard = "^2.18.0"
huggingface-hub = "^0.26.2"
tyro = "^0.8.14"
torch = "^2.5.1"
stable-baselines3 = "^2.3.2"
numpy = "^1.21.6"
tenacity = "^9.0.0"
mujoco = "2.3.3"


[tool.poetry.group.dev.dependencies]
black = "^24.10.0"
wandb = "^0.18.7"
moviepy = "^2.1.1"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"