ctm-dqn/pyproject.toml

32 lines
709 B
TOML

[[tool.uv.index]]
url = "https://pypi.tuna.tsinghua.edu.cn/simple"
default = true
[project]
name = "ctm"
version = "0.1.0"
description = "Multi-model Dynamic Speed Limit Control with Cell Transmission Model"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
"pyyaml>=6.0",
"tqdm>=4.65.0",
"eclipse-sumo>=1.20.0",
"torch==2.4.1+cu124",
"numpy>=2.4.3",
"matplotlib>=3.10.8",
"stable-baselines3>=2.7.1",
]
[[tool.uv.index]]
name = "tuna"
url = "https://pypi.mirrors.ustc.edu.cn/simple/"
default = true
[[tool.uv.index]]
name = "pytorch-cu124"
url = "https://mirrors.nju.edu.cn/pytorch/whl/cu124"
explicit = true
[tool.uv.sources]
torch = { index = "pytorch-cu124" }