[project]
name = "lab-base"
version = "0.1.0"
description = "Base reproducible Python environment for research labs"
requires-python = ">=3.11"
dependencies = ["numpy>=2.3.2", "manim-slides"]

[tool.pixi.project]
platforms = ["linux-64", "osx-arm64", "win-64"]
channels = ["conda-forge"]

[tool.pixi.dependencies]
python = "3.11.*"
ffmpeg = "*"
cairo = "*"
pango = "*"
graphviz = "*"
pkg-config = ">=0.29.2,<0.30"
pycairo = "*"
zlib = "*"
numpy = "*"
scipy = "*"
matplotlib = "*"
jupyter = "*"
ipykernel = "*"
tqdm = "*"
manim = "*"


# Features for torch
[tool.pixi.feature.torch-cuda.pypi-dependencies]
torch = "*"
torchvision = "*"
torchaudio = "*"

[tool.pixi.feature.torch-cpu-or-mps.pypi-dependencies]
torch = "*"
torchvision = "*"
torchaudio = "*"

# CUDA system deps (Linux/Windows only)
[tool.pixi.target.linux-64.dependencies]
cuda-toolkit = ">=12.4"
cudnn = ">=9,<10"

[tool.pixi.target.win-64.dependencies]
cuda-toolkit = ">=12.4"
cudnn = ">=9,<10"

[tool.pixi.tasks]
lab = "jupyter lab"