File size: 967 Bytes
6a51385 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | [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
|