[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "flexibrain"
version = "0.1.0"
description = "Flexible-resolution fMRI Mamba-JEPA pretraining and downstream classification"
readme = "README.md"
requires-python = ">=3.10,<3.11"
license = {file = "LICENSE"}
dependencies = [
  "numpy==1.26.2",
  "pandas==2.1.3",
  "PyYAML==6.0.1",
  "nibabel==5.3.2",
  "scikit-learn==1.3.2",
  "timm==0.4.12",
  "einops==0.7.0",
  "packaging>=23.2",
  "triton==2.1.0",
]

[project.optional-dependencies]
cuda = [
  "torch==2.1.2",
  "torchvision==0.16.2",
  "torchaudio==2.1.2",
  "causal-conv1d==1.2.2.post1",
  "mamba-ssm==2.0.3",
  "flash-attn==2.5.8",
]
preprocess = [
  "nilearn==0.10.3",
  "tqdm==4.66.1",
]

[project.scripts]
flexibrain = "flexibrain.cli:main"

[tool.setuptools]
packages = {find = {include = ["flexibrain*", "mamba_ssm*"]}}
include-package-data = true