gpudrive/pyproject.toml at main · Emerge-Lab/gpudrive · 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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
[build-system]
requires = [
    "setuptools>=42",
    "scikit-build>=0.13",
    "cmake>=3.18",
    "ninja",
]
build-backend = "madrona_py_build"
backend-path = ["external/madrona/py"]
wheel-directory = "build"

[project]
name = "gpudrive"
version = "0.4.0"
authors = [
    {name = "Saman Kazemkheni", email = "skazemkhani@gmail.com"},
    {name = "Aarav Pandya", email = "pandya.aarav.97@gmail.com"},
    {name = "Daphne Cornelisse", email = "cor.daphne@gmail.com"},
    {name = "Eugene Vinitsky", email ="vinitsky.eugene@gmail.com"},
]
readme = "README.md"
description = "A GPU-accelerated, multi-agent driving simulator"
requires-python = ">=3.11"
dependencies = [
    "numpy>=1.26.4,<2",
    "gymnasium",
    "pygame",
    "matplotlib==3.9",
    "pandas",
    "python-box==7.2.0",
    "typer",
    "pyyaml",
    "mediapy",
    "wandb",
    "torch>=2.2.1",
    "seaborn",
    "safetensors",
    "python-box",
    "tqdm",
    "jax",
    "huggingface_hub",
    "cmake==4.0.0",
]

[dependency-groups]
pufferlib = [
    "pufferlib>=2.0.6,<3",
]
sb3 = [
    "stable-baselines3==2.3.2",
]
test = [
    "pytest>=8.2.1",
]
vbd = [
    "lightning",
    "jaxlib==0.5.3",  # see https://github.com/Emerge-Lab/gpudrive/issues/464
    "waymo-waymax @ git+https://github.com/waymo-research/waymax.git@main",
]

[tool.uv]
default-groups = "all"
cache-dir = "./.uv_cache"

# Use the CUDA index for torch on Linux
[[tool.uv.index]]
name = "pytorch-cuda"
url = "https://download.pytorch.org/whl/cu124"
explicit = true

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

[tool.uv.sources]
torch = [
    { index = "pytorch-cuda", marker = "platform_system == 'Linux'" },
    { index = "pytorch-cpu", marker = "platform_system != 'Linux'" }
]

[tool.madrona.packages.madrona_gpudrive]
ext-only = true
ext-out-dir = "build"

[tool.madrona.packages.gpudrive]
path = "gpudrive"

[project.packages]
include = [
    "baselines",
    "baselines/*",
]