pyproject.toml

[project]
name = "mini-nvs-solver"
version = "0.1.0"
description = "Add a short description here"
authors = [{ name = "Pablo Vela", email = "[email protected]" }]
requires-python = ">= 3.10"
dependencies = [
    "monopriors>=0.1.0,<0.2",
    "mmcv-lite>=2.2.0,<3",
    "calibur>=0.0.1,<0.0.2",
    "gradio-rerun>=0.0.4,<0.0.5", "hf-transfer>=0.1.8,<0.2",
]

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

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

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

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

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


[tool.pixi.tasks]
_install-wheel = "python -m pip install $WHL_PATH --force-reinstall"
_build-wheel = "python -m pip install build && 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.tasks.rerun-demo]
cmd = """
    python tools/rerun_demo.py --headless --connect
    """
description = "Runs Rerun demo"

[tool.pixi.feature.spaces.tasks.app]
cmd = """
    PYTORCH_JIT=0 python gradio_app.py
    """
description = "Runs Gradio frontend for depth comparison"
depends_on = ["_install-wheel"]

[tool.pixi.pypi-dependencies]
mini-nvs-solver = { path = ".", editable = true }

[tool.pixi.dependencies]
python = "3.10.*"
pip = ">=24.0,<25"
cuda = { version = "*", channel = "nvidia/label/cuda-11.8.0" }
pytorch-cuda = { version = "11.8.*", channel = "pytorch" }
pytorch = { version = "2.3.*", channel = "pytorch" }
torchvision = { version = "0.18.*", channel = "pytorch" }
xformers = { version = ">=0.0.26.post1,<0.0.27", channel = "xformers" }
numpy = "<2.0.0"
jaxtyping = ">=0.2.30,<0.3"
einops = ">=0.8.0,<0.9"
transformers = ">=4.42.3,<4.43"
diffusers = ">=0.29.2,<0.30"
accelerate = ">=0.32.0,<0.33"
pyserde = ">=0.19.3,<0.20"
trimesh = ">=4.4.1,<4.5"
icecream = ">=2.1.3,<2.2"
timm = ">=1.0.7,<1.1"
wandb = ">=0.16.6,<0.17"
tqdm = ">=4.66.4,<4.67"
beartype = ">=0.18.5,<0.19"
pyperclip = "==1.8.2"
zip = ">=3.0,<4"
gradio = ">=4.40.0,<5"
rerun-sdk = ">=0.17.0,<0.18"

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


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

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