[workspace]

authors = ["Rishabh Ranjan <rishabhranjan0207@gmail.com>"]

channels = ["conda-forge"]

name = "relbench"

platforms = ["linux-64", "osx-arm64"]

version = "3.0.1"

[tasks]

[dependencies]

uv = "*"

python = "==3.10"

[pypi-options]

index-url = "https://pypi.org/simple"

[pypi-dependencies]

relbench = { path = ".", editable = true, extras = ["schema", "example", "dev"] }

# Two environments select the accelerator for torch + the matching pyg-lib wheels.

# `default` (= cpu) targets this dev box and CI, which have no GPU. `gpu` pulls CUDA

# (cu128) builds; run the examples on a GPU node with `pixi run -e gpu python examples/...`.

[feature.cpu.pypi-options]

find-links = [{ url = "https://data.pyg.org/whl/torch-2.9.0+cpu.html" }]

[feature.cpu.pypi-dependencies]

torch = { version = "==2.9.1", index = "https://download.pytorch.org/whl/cpu" }

pyg-lib = "*"

[feature.gpu]

platforms = ["linux-64"]

[feature.gpu.pypi-options]

find-links = [{ url = "https://data.pyg.org/whl/torch-2.9.0+cu128.html" }]

[feature.gpu.pypi-dependencies]

torch = { version = "==2.9.1", index = "https://download.pytorch.org/whl/cu128" }

pyg-lib = "*"

[environments]

default = ["cpu"]

gpu = ["gpu"]

[tasks.test]

cmd = "pytest test"