adversarial-simulation/pyproject.toml at master · 0xMihir/adversarial-simulation · 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
94
95
96
97
98
99
100
101
102
103
104
105
[project]
name = "adversarial-simulation"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12,<3.13"
dependencies = [
    "faro",
    "schema",
    "synthetic",
    "shapely>=2.0.0",
    "crc32c>=2.3",
    "noise>=1.2.2",
    "accelerate>=1.12.0",
    "grpcio-tools>=1.80.0",
    "ipykernel>=7.1.0",
    "ipympl>=0.9.8",
    "ipywidgets>=8.1.8",
    "jupyter>=1.1.1",
    "jupyter-mcp-tools>=0.1.6",
    "lxml>=6.0.2",
    "matplotlib>=3.10.7",
    "mplcursors>=0.7",
    "numba>=0.63.1",
    "numpy>=2.3.5",
    "openai>=2.15.0",
    "pandas>=2.3.3",
    "protobuf>=6.33.2",
    "pyclothoids>=0.2.0",
    "python-dotenv>=1.2.1",
    "rtree>=1.4.1",
    "scipy>=1.16.3",
    "torch==2.12.0",
    "torchvision>=0.24.1",
    "transformers>=4.57.3",
    "fastapi>=0.135.2",
    "uvicorn[standard]>=0.42.0",
    "torch-geometric>=2.8.0",
    "models",
    "wandb>=0.27.2",
]

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

[[tool.uv.index]]
name = "cumm-spconv"
url = "https://ratharog.github.io/cumm-spconv/"
explicit = true

[tool.ruff]
extend-exclude = ["*.md"]

[dependency-groups]
dev = [
    "pytest>=9.0.3",
]

[tool.uv.sources]
faro = { path = "./preprocessing", editable = true }
schema = { path = "./schema", editable = true }
synthetic = { path = "./synthetic", editable = true }
models = { path = "./models", editable = true }
# cumm/spconv: editable path installs from local repos — git-source builds
# omit the tensorview headers (needed at JIT time), and Blackwell (sm_121)
# isn't supported for CUMM_DISABLE_JIT=1 precompiled kernels in 0.8.2.
cumm = [
  { path = "/home/mpatil/Programming/Repos/cumm", editable = true, marker = "platform_machine == 'aarch64'" },
]
spconv = [
  { path = "/home/mpatil/Programming/Repos/spconv", editable = true, marker = "platform_machine == 'aarch64'" },
]

torch = [
  { index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]
torchvision = [
  { index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]

[tool.uv]
no-build-isolation-package = ["spconv", "torch-scatter", "torch-cluster", "flash-attn", "flash-attn-3"]
override-dependencies = ["cumm>=0.7.11"]
# Only the dev box (aarch64 Jetson) and Anvil (x86_64 linux) are real targets;
# restricting this avoids uv's universal resolver trying (and failing) to
# also cover untested platforms like macOS/Windows x86_64 for cuda-suffixed
# packages (spconv-cu130 etc.) that only ship linux/win wheels.
environments = [
    "platform_machine == 'aarch64' and sys_platform == 'linux'",
    "platform_machine == 'x86_64' and sys_platform == 'linux'",
]

[tool.uv.extra-build-dependencies]
spconv = ["pccm>=0.4.16", "ccimport>=0.4.4", "pybind11>=2.6.0", "fire"]
flash-attn = ["packaging", "ninja", "wheel", "setuptools", "psutil"]
flash-attn-3 = ["packaging", "ninja", "wheel", "setuptools", "psutil", "einops"]

[tool.basedpyright]
pythonVersion = "3.12"
typeCheckingMode = "standard"
venvPath = "."
venv = ".venv"
extraPaths = [".", "../../Repos/spconv", "../../Repos/cumm"]