numpy<2,>=1.17.5
openai
pandas<2,>=1
orion-ml<0.8,>=0.5
scikit-learn<1.2,>=0.22
tiktoken
transformers
torch>=1.4
accelerate
sentencepiece

[dev]
bumpversion>=0.5.3
pip>=9.0.1
watchdog>=0.8.3
docutils<0.18,>=0.12
m2r2<0.3,>=0.2.5
nbsphinx<0.7,>=0.5.0
Sphinx<3.3,>=3
pydata-sphinx-theme<0.5
markupsafe<2.1.0
ipython<9,>=6.5
Jinja2<3,>=2
alabaster<=0.7.12
sphinxcontrib-applehelp<1.0.8
sphinxcontrib-devhelp<1.0.6
sphinxcontrib-htmlhelp<2.0.5
sphinxcontrib-serializinghtml<1.1.10
sphinxcontrib-qthelp<1.0.7
flake8<4,>=3.7.7
isort<5,>=4.3.4
autoflake<2,>=1.2
autopep8<2,>=1.4.3
importlib-metadata<5
twine<4,>=1.10.0
wheel>=0.30.0
coverage<6,>=4.5.1
tox<4,>=2.9.1
invoke
pytest>=3.4.2
pytest-cov>=2.6.0
rundoc<0.5,>=0.4.3

[test]
pytest>=3.4.2
pytest-cov>=2.6.0
rundoc<0.5,>=0.4.3
