mario-openenv/pyproject.toml at main · 3xcaffeine/mario-openenv · GitHub

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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
[build-system]
requires = ["uv_build>=0.9.5,<0.10.0"]
build-backend = "uv_build"

[project]
name = "mario-openenv"
version = "0.1.0"
description = "Multi-approach reinforcement learning framework for Super Mario Bros, featuring traditional PPO, LLM-based GRPO, and OpenEnv-compatible environment wrappers"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "bitsandbytes>=0.48.1",
    "fastapi[standard]>=0.120.0",
    "gym-super-mario-bros",
    "gymnasium[other]>=1.2.1",
    "ipykernel>=7.1.0",
    "ipywidgets>=8.1.7",
    "modal>=1.2.1",
    "nes-py",
    "numba>=0.62.1",
    "numpy>=2.3.4",
    "opencv-python>=4.11.0.86",
    "opencv-python-headless>=4.11.0.86",
    "openenv-core>=0.1.0",
    "torch>=2.9.0",
    "torchao==0.14.1",
    "torchvision>=0.23.0",
    "tqdm>=4.67.1",
    "trackio>=0.7.0",
    "transformers==4.56.2",
    "triton>=3.5.0",
    "triton-kernels",
    "trl==0.22.2",
    "unsloth-zoo[base]",
    "unsloth[base]",
]

[tool.uv.build-backend]
module-name = "mario_env"
module-root = ""

[tool.uv.sources]
gym-super-mario-bros = { git = "https://github.com/Leirbag-gabrieL/gym-super-mario-bros-fixed", rev = "f4494deb202dd216a7552e69b73be9e7e80b2439" }
nes-py = { git = "https://github.com/Leirbag-gabrieL/nes-py-fixed", rev = "2773959fb8b34f0a42d4b0a51966252d35208863" }
torch = [
    { index = "pytorch-cpu", extra = "cpu" },
    { index = "pytorch-cu128", extra = "gpu" },
]
unsloth-zoo = { git = "https://github.com/unslothai/unsloth-zoo" }
unsloth = { git = "https://github.com/unslothai/unsloth" }
triton-kernels = { git = "https://github.com/triton-lang/triton.git", subdirectory = "python/triton_kernels", rev = "05b2c186c1b6c9a08375389d5efe9cb4c401c075" }

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true

[project.optional-dependencies]
gpu = ["torch>=2.9.0"]
cpu = ["torch>=2.9.0"]

[tool.uv]
conflicts = [[{ extra = "cpu" }, { extra = "gpu" }]]

[tool.pytest.ini_options]
filterwarnings = [
    "ignore::pytest.PytestAssertRewriteWarning",
]

[dependency-groups]
dev = [
    "opencv-python>=4.11.0.86",
]