tokeye/pyproject.toml at main · PlasmaControl/tokeye · 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
[build-system]
requires = ["uv_build"]
build-backend = "uv_build"

[project]
name = "tokeye"
version = "0.12.0"
description = "Automatic classification and localization of fluctuating signals in spectrograms"
readme = "README.md"
requires-python = ">=3.13"
dependencies = [
    "numpy",
    "scipy",
    "matplotlib",
    # PyTorch 2.11 dropped Volta/V100 (sm_70) from its CUDA wheels; DIII-D's
    # omega GPU partition is V100 (gres=gpu:v100). 2.10 is the last release
    # that runs there on GPU. Lift when the deploy targets newer hardware.
    "torch<2.11",
    "torchvision",  # follows torch via torchvision's own exact torch pin
    "torchinfo",
    "omegaconf",
    "pydantic",
    "huggingface-hub>=0.30",
    "tqdm",
    "pyyaml",
]

[project.optional-dependencies]
app = [
    # gradio 6 removes Image(show_download_button=...) (used in app/__main__.py)
    # and moves theme=/css= from Blocks() to launch() — the app is 5.x-only.
    "gradio>=5.49,<6",
]
eigspec = [
    "scikit-learn",
]
train = [
    "ipykernel",
    "ipywidgets",
    "h5py",
    "wavio",
    "joblib",
    "lightning",
    "torchaudio<2.11",  # match torch<2.11 (2.11 dropped its torch-version metadata)
    "torchmetrics",
    "scikit-learn",
    "scikit-image",
    "pybaselines",
    "tifffile",
    "pandas",
    "tables",
    "kneed",
    "threadpoolctl",
]

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

[tool.pytest.ini_options]
minversion = "9.0"
addopts = ["-ra", "-q"]
testpaths = [
    "tests",
]

[tool.uv]
package = true

[tool.uv.build-backend]
module-name = "tokeye"
module-root = "src"

[dependency-groups]
dev = [
    "ty",
    "pytest",
    "pytest-cov",
    "pytest-xdist",
    "ruff",
    "pre-commit",
]
train = [
    "ipykernel",
    "ipywidgets",

    "h5py",
    "wavio",
    "joblib",

    "lightning",
    "torchaudio<2.11",  # match torch<2.11 (2.11 dropped its torch-version metadata)
    "torchmetrics",

    "scikit-learn",
    "scikit-image",

    "pybaselines",
    "tifffile",
    "pandas",
    "tables",

    "kneed",
    "threadpoolctl",
]