fmri-fm/pyproject.toml at main · MedARC-AI/fmri-fm · GitHub

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
45
46
47
48
49
50
51
52
53
54
55
[build-system]
requires = ["setuptools>=64", "setuptools_scm[toml]>=8"]
build-backend = "setuptools.build_meta"

[project]
name = "fmri-fm"
dynamic = ["version"]
description = "MedARC fMRI foundation model"
readme = "README.md"
requires-python = ">=3.11"
dependencies = [
    "datasets",
    "einops",
    "huggingface_hub",
    "ipython",
    "jaxtyping",
    "jupyter",
    "matplotlib",
    "nibabel",
    "nsdcode",
    "numpy",
    "omegaconf",
    "pre-commit",
    "pytest",
    "pycortex",
    "scikit-learn",
    "scipy",
    "timm",
    "torch==2.8.0",
    "torchvision",
    "tqdm",
    "wandb",
    "webdataset",
    "cloudpathlib[s3]",
    "lightning-sdk",
]

[project.optional-dependencies]

eval = [
    "fmri-fm-eval"
]

[tool.setuptools_scm]
version_file = "src/_version.py"

[tool.uv.sources]
fmri-fm-eval = { git = "https://github.com/MedARC-AI/fmri-fm-eval.git" }
nsdcode = { git = "https://github.com/cvnlab/nsdcode.git", rev = "bfd36a503cc90a3eb3ebfd69b269403f7e186924" }

[tool.ruff]
line-length = 100

[tool.ruff.lint]
ignore = ["F722"]