torch>=2.2.2
scikit-learn>=1.2.2
tensordict>=0.3.2
unipercept

[docs]
sphinx
sphinx-rtd-theme
numpydoc

[notebooks]
jupyterlab
ipywidgets
ipykernel
ipython-autotime
matplotlib>=3.8.4
seaborn>=0.13.2

[qa]
black>=24.3.0
bandit[toml]
check-manifest
flake8>=7.0.0
flake8-bugbear
flake8-docstrings
flake8-formatter_junit_xml
flake8-import-conventions
flake8-unused-arguments
flake8-comprehensions
flake8-errmsg
flake8-logging-format
flake8-pie
flake8-pyi
flake8-pytest-style
flake8-return
flake8-2020
flake8-executable
flake8-simplify
flake8-black
flake8-pyproject
pre-commit
pygrep
isort>=5.13.2
refurb
pylint>=3.0.3
pylint_junit
shellcheck-py
pylint
mypy>=1.9.0
autoflake
pydocstyle
pyre-check
pydocstringformatter
pyrefact
pyflakes
mccabe

[tests]
hypothesis>=6.100.1
pytest>=8.1.1
pytest-sugar
pytest-xdist
pytest-benchmark
pytest-cov
pytest-mock
pytest-runner
pytest-github-actions-annotate-failures
