pyproject.toml

[project]
name = "dehazing"
version = "0.1.0"
requires-python = ">=3.11, <3.14"
dependencies = [
  # Using >= allows both "2.7.0" (if exact) and "2.7.0.dev..." (nightly)
  "torch==2.7.0",
  "torchvision==0.22.0",
  "torchaudio==2.7.0",
  # Mamba_ssm support only this version of transformers
  "transformers==4.57.3",
  "filelock>=3.13.1",
  "fsspec>=2024.0.0",
  "jinja2>=3.1.0",
  "markupsafe>=2.1.0",
  "mpmath>=1.3.0",
  "networkx>=3.0",
  "numpy>=1.26.0",
  "pillow>=10.0.0",
  "sympy>=1.12",
  "triton>=3.0.0",
  "typing-extensions>=4.9.0",
  "torchdiffeq>=0.2.5",
  "torchinfo>=1.8.0",
  "torchmetrics>=1.8.2",
  "pandas>=2.2.0",
  "matplotlib>=3.8.0",
  "seaborn>=0.13.0",
  "tqdm>=4.66.0",
  "kaggle>=1.6.0",
  "einops>=0.8.0",
  "jupyter>=1.0.0",
  "yacs>=0.1.8",
  "tensorboard>=2.15.0",
  "thop>=0.1.1.post2209072238",
  "rich>=13.7.0",
  # --- BUILD DEPENDENCIES ---
  # These are critical for the fallback "Source Build" to work on Mac/Windows
  "ninja>=1.11.1",
  "packaging>=23.2",
  # The main libraries
  "mamba_ssm==2.2.6.post3",
  "causal_conv1d==1.5.4",
  "lpips>=0.1.4",
  "opencv-python>=4.11.0.86",
  "huggingface-hub>=0.36.2",
]

# --- UV Configuration ---

[tool.uv]
prerelease = "allow"
# Ensure build environment sees the Nightly Torch
build-constraint-dependencies = ["torch==2.7.0", "ninja", "packaging"]

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true 

[tool.uv.sources]
torch = [{ index = "pytorch-cu128" }]
torchvision = [{ index = "pytorch-cu128" }]
torchaudio = [{ index = "pytorch-cu128" }]

# --- SMART FALLBACK LOGIC ---
# Strategy: Use specific markers to direct Linux Users to Wheels, and everyone else to Git.

causal-conv1d = [
    # PATH A: Binary Wheels (Linux x86_64 ONLY) - ABI=True
    { url = "https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.5.4/causal_conv1d-1.5.4+cu12torch2.7cxx11abiTRUE-cp311-cp311-linux_x86_64.whl", marker = "sys_platform == 'linux' and platform_machine == 'x86_64' and python_version == '3.11'" },
    { url = "https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.5.4/causal_conv1d-1.5.4+cu12torch2.7cxx11abiTRUE-cp312-cp312-linux_x86_64.whl", marker = "sys_platform == 'linux' and platform_machine == 'x86_64' and python_version == '3.12'" },
    { url = "https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.5.4/causal_conv1d-1.5.4+cu12torch2.7cxx11abiTRUE-cp313-cp313-linux_x86_64.whl", marker = "sys_platform == 'linux' and platform_machine == 'x86_64' and python_version == '3.13'" },
    
    # PATH B: Fallback to Source (Mac, Windows, or ARM Linux)
    { git = "https://github.com/Dao-AILab/causal-conv1d/", tag = "v1.5.4", marker = "sys_platform != 'linux' or platform_machine != 'x86_64'" }
]

mamba-ssm = [
    # PATH A: Binary Wheels (Linux x86_64 ONLY) - ABI=True
    { url = "https://github.com/state-spaces/mamba/releases/download/v2.2.6.post3/mamba_ssm-2.2.6.post3+cu12torch2.7cxx11abiTRUE-cp311-cp311-linux_x86_64.whl", marker = "sys_platform == 'linux' and platform_machine == 'x86_64' and python_version == '3.11'" },
    { url = "https://github.com/state-spaces/mamba/releases/download/v2.2.6.post3/mamba_ssm-2.2.6.post3+cu12torch2.7cxx11abiTRUE-cp312-cp312-linux_x86_64.whl", marker = "sys_platform == 'linux' and platform_machine == 'x86_64' and python_version == '3.12'" },
    { url = "https://github.com/state-spaces/mamba/releases/download/v2.2.6.post3/mamba_ssm-2.2.6.post3+cu12torch2.7cxx11abiTRUE-cp313-cp313-linux_x86_64.whl", marker = "sys_platform == 'linux' and platform_machine == 'x86_64' and python_version == '3.13'" },

    # PATH B: Fallback to Source (Mac, Windows, or ARM Linux)
    { git = "https://github.com/state-spaces/mamba", tag = "v2.2.6.post3", marker = "sys_platform != 'linux' or platform_machine != 'x86_64'" }
]

[dependency-groups]
dev = [
    "ipykernel>=7.1.0",
]