numpy>=1.24
pandas>=2.0
scipy>=1.10
statsmodels>=0.14
matplotlib>=3.7

[all]
conformal-oracle[benchmarks,tsfm_all]

[benchmarks]
arch>=6.0

[chronos]
chronos-forecasting>=1.5
torch>=2.0
transformers>=4.30

[dev]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.1
mypy>=1.5
scikit-learn>=1.0

[lag_llama]
torch>=2.0
gluonts>=0.14
huggingface-hub>=0.20
einops>=0.6

[moirai]
uni2ts>=0.1
torch>=2.0
einops>=0.6
gluonts>=0.14
huggingface-hub>=0.20

[timesfm]
timesfm>=1.2
torch>=2.0

[tsfm_all]
chronos-forecasting>=1.5
timesfm>=1.2
uni2ts>=0.1
torch>=2.0
transformers>=4.30
gluonts>=0.14
huggingface-hub>=0.20
einops>=0.6
