[project]
name = "mini-dust3r"
version = "0.1.1"
description = "Add a short description here"
authors = ["pablovela5620 <pablovela5620@gmail.com>"]
channels = ["nvidia/label/cuda-11.8.0", "nvidia", "conda-forge", "pytorch"]
platforms = ["linux-64", "osx-arm64"]

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

[tasks]
build-croco = { cmd = "python setup.py build_ext --inplace", cwd = "mini_dust3r/croco/curope/", outputs=["lib.linux-x86_64-cpython-311/curope.cpython-311-x86_64-linux-gnu.so"]}
download-checkpoint = {cmd = "ls checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth || wget -P checkpoints/ https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth", outputs=["checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth"]}
post-install = { cmd = "pwd", depends_on = ["download-checkpoint", "build-croco"]}

rerun-demo = "PYTORCH_ENABLE_MPS_FALLBACK=1 python tools/rerun_demo.py"
gradio-demo = "python tools/app.py"

dev-install = "python -m pip install build twine"
build-wheel = { cmd = "python -m build", depends_on = ["dev-install"]}

[dependencies]
python = "3.11.*"
pip = ">=23.3.2,<23.4"
pytorch = {version = ">=2.2.0,<2.3", channel="pytorch"}
torchvision = {version = ">=0.17.0,<0.18", channel="pytorch"}
matplotlib = ">=3.8.3,<3.9"
einops = ">=0.7.0,<0.8"
tqdm = ">=4.66.2,<4.67"
scipy = ">=1.12.0,<1.13"
trimesh = ">=4.2.0,<4.3"
tensorboard = ">=2.16.2,<2.17"
pyglet = ">=1.5.27,<1.6"
jaxtyping = ">=0.2.28,<0.3"
gradio = ">=4.29.0,<4.30"
rerun-sdk = ">=0.15.1,<0.16"

[target.linux-64.dependencies]
cuda = {version = "*", channel="nvidia/label/cuda-11.8.0"}
pytorch-cuda = {version = "11.8.*", channel="pytorch"}

[pypi-dependencies]
mini-dust3r = { path = ".", editable = true}