pandas>=2.2.0
scikit-learn
transformers[torch]>=4.38.0
datasets
deprecated
urllib3>=1.26.19
numpy<3
torch<3

[all]
granite-tsfm[dev,notebooks,testing]

[demos]
granite-tsfm[notebooks]
streamlit
altair<5
plotly
streamlit-aggrid
kaleido
toml

[dev]
pre-commit
granite-tsfm[testing]
ruff==0.5.7

[notebooks]
jupyter
matplotlib
datasets
ipywidgets
plotly
kaleido
tensorboard

[ogv]
datasets
plotly
kaleido

[testing]
pytest
pytest-coverage
granite-tsfm[notebooks]
parameterized
