[project]
name = "template_project_alpha"
version = "0.1.0"
description = "a basic template project that has the ability to complete most of ML task with PyTorch and MMCV"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
  "anyio>=4.9.0",
  "awkward>=2.8.5",
  "beartype>=0.21.0",
  "click>=8.2.1",
  "jaxtyping>=0.3.2",
  "loguru>=0.7.3",
  "mmcv",
  "mmdet>=3.3.0",
  "mmpose>=1.3.2",
  "nats-py>=2.10.0",
  "numba>=0.61.2",
  "nvidia-nvimgcodec-cu12>=0.5.0.13",
  "opencv-python>=4.12.0.88",
  "orjson>=3.10.18",
  "pyarrow>=20.0.0",
  "pydantic>=2.11.7",
  "pytorch3d",
  "redis>=6.2.0",
  "result>=0.17.0",
  "tomli>=2.2.1",
  "tomli-w>=1.2.0",
  "torch>=2.7.0",
  "torchvision>=0.22.0",
  "ultralytics>=8.3.166",
  "xtcocotools",
]

[dependency-groups]
dev = ["jupyterlab>=4.4.4", "pip", "psutil>=7.0.0", "setuptools"]

[tool.uv]
no-build-isolation-package = ["chumpy", "xtcocotools"]

[tool.uv.sources]
torch = [
  { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
  { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
]
torchvision = [
  { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
  { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
]
# built with cu128
mmcv = { path = "misc/mmcv-2.2.0-cp312-cp312-linux_x86_64.whl" }
xtcocotools = { path = "misc/xtcocotools-1.14.3-cp312-cp312-linux_x86_64.whl" }
pytorch3d = { path = "misc/pytorch3d-0.7.8-cp312-cp312-linux_x86_64.whl" }

[[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