numpy>=1.24
scipy>=1.13
pandas>=2.2.2
scikit-learn>=1.4.2
matplotlib>=3.8.4
tqdm>=4.60
pyyaml>=6.0
omegaconf<3,>=2.3
safetensors>=0.4
huggingface_hub>=0.20

[:platform_system != "Darwin" or platform_machine != "x86_64"]
torch>=2.3

[:platform_system == "Darwin" and platform_machine == "x86_64"]
torch>=2.0
numpy<2

[:platform_system == "Darwin" and python_version < "3.11"]
scipy!=1.15.*

[all]
black
flake8
hydra-core<2,>=1.3.2
hydra-list-sweeper<2,>=1.1.0
ipykernel
isort
jupyter
mantis-tsfm<2,>=1.0.0
neurokit2>=0.2.10
notebook>=7.0
pytest-cov
pytest>=8.0
pytorch-lightning<3,>=2.6
tables>=3.10
torchmetrics>=1.0
torchvision>=0.15
umap-learn>=0.5.4
wandb>=0.19.9

[dev]
pytest>=8.0
pytest-cov
black
flake8
isort

[mantis]
mantis-tsfm<2,>=1.0.0

[notebook]
jupyter
notebook>=7.0
ipykernel

[signal]
neurokit2>=0.2.10

[train]
pytorch-lightning<3,>=2.6
torchmetrics>=1.0
torchvision>=0.15
hydra-core<2,>=1.3.2
hydra-list-sweeper<2,>=1.1.0
wandb>=0.19.9
tables>=3.10
neurokit2>=0.2.10
umap-learn>=0.5.4

[viz]
umap-learn>=0.5.4
