[project]
name = "zero_spaces_template"
version = "0.1.0"
description = "Add a short description here"
authors = [{ name = "Pablo Vela", email = "pablovela5620@gmail.com" }]
requires-python = ">= 3.11"
dependencies = [
    "mmcv-lite>=2.2.0,<2.3",
    "gradio-rerun==0.0.4",
    "build>=1.2.1,<2",
    "hf-transfer>=0.1.8,<0.2",
]

[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
include = ["zero_spaces_template*"]

[tool.pixi.project]
channels = ["nvidia", "nvidia/label/cuda-11.8.0", "conda-forge", "pytorch"]
platforms = ["linux-64"]

[tool.pixi.system-requirements]
libc = { family = "glibc", version = "2.31" }

[tool.pixi.activation]
scripts = [".pixi.sh"]

[tool.pixi.pypi-options]
no-build-isolation = ["gsplat"]

[tool.pixi.tasks]
_install-wheel = "uv pip install ./$WHL_PATH --force-reinstall"
_build-wheel = "python -m build"
upload-hf = { cmd = "python tools/upload_to_hf.py", depends_on = [
    "_build-wheel",
] }

[tool.pixi.tasks.app]
cmd = """
    python tools/gradio_app.py
    """
description = "Runs Gradio frontend"

[tool.pixi.tasks.dev-app]
cmd = """
    gradio tools/gradio_app.py
    """
description = "Runs Gradio frontend for hotreload"

[tool.pixi.feature.spaces.tasks.app]
cmd = """
    python gradio_app.py
    """
description = "Runs Gradio frontend in huggingface spaces"
depends_on = ["_install-wheel"]

[tool.pixi.pypi-dependencies]
gsplat = { git = "https://github.com/nerfstudio-project/gsplat.git", rev = "2d456983e345ebbfa65ab3b119a02113b84b4cf6" }
zero_spaces_template = { path = ".", editable = true }

[tool.pixi.dependencies]
python = "3.11.*"
pytorch = { version = "2.3.1", channel = "pytorch" }
torchvision = { version = "0.18.1", channel = "pytorch" }
cuda = { version = "*", channel = "nvidia/label/cuda-11.8.0" }
cuda-nvcc = { version = "11.8.*", channel = "nvidia/label/cuda-11.8.0" }
pytorch-cuda = { version = "11.8.*", channel = "pytorch" }
beartype = ">=0.18.5,<0.19"
rerun-sdk = "==0.17.0"
jaxtyping = ">=0.2.33,<0.3"
gradio = ">=4.42.0,<5"
uv = ">=0.4.15,<0.5"
setuptools = ">=74.1.2,<75"

[tool.pixi.feature.spaces.pypi-dependencies]
spaces = "==0.30.2"


[tool.pixi.environments]
spaces = { features = ["spaces"], solve-group = "default" }

[tool.ruff]
ignore = [
    "F722", # Forward annotation false positive from jaxtyping. Should be caught by pyright.
]