numpy
scikit-learn
scipy
torch>=1.10.0
tensorboard
pandas<2.0.0
pygrinder>=0.2
tsdb>=0.2
h5py

[basic]
numpy
scikit-learn
pandas<2.0.0
torch>=1.10.0
tensorboard
scipy
h5py
tsdb>=0.2
pygrinder>=0.2

[dev]
black
flake8
pre-commit
jupyterlab
numpy
scikit-learn
pandas<2.0.0
torch>=1.10.0
tensorboard
scipy
h5py
tsdb>=0.2
pygrinder>=0.2
torch-geometric
torch-scatter
torch-sparse
nni
pytest-xdist
pytest-cov
furo
sphinx
sphinxcontrib-bibtex
sphinxcontrib-gtagjs
sphinx-autodoc-typehints

[doc]
furo
sphinx
sphinxcontrib-bibtex
sphinxcontrib-gtagjs
sphinx-autodoc-typehints

[full]
numpy
scikit-learn
pandas<2.0.0
torch>=1.10.0
tensorboard
scipy
h5py
tsdb>=0.2
pygrinder>=0.2
torch-geometric
torch-scatter
torch-sparse
nni

[optional]
torch-geometric
torch-scatter
torch-sparse
nni

[test]
pytest-xdist
pytest-cov
