numpy>=1.21.0
tqdm>=4.27.0

[all]
cappr[hf,llama-cpp,openai]

[demos]
cappr[hf-dev,llama-cpp-dev,openai]
datasets>=2.10.0
jupyter>=1.0.0
matplotlib>=3.7.3
pandas>=1.5.3
scikit-learn>=1.2.2

[dev]
cappr[demos,hf-dev,llama-cpp-dev,openai]
docutils<0.19
pre-commit>=3.5.0
pydata-sphinx-theme>=0.13.1
pytest>=7.2.1
pytest-cov>=4.0.0
pytest-sugar>=1.0.0
ruff>=0.3.0
sphinx>=6.1.3
sphinx-copybutton>=0.5.2
sphinx-togglebutton>=0.3.2
sphinxcontrib-napoleon>=0.7
twine>=4.0.2

[hf]
transformers[torch]>=4.31.0

[hf-dev]
transformers[torch]>=4.35.0
huggingface-hub>=0.16.4
sentencepiece>=0.1.99

[llama-cpp]
llama-cpp-python>=0.2.11

[llama-cpp-dev]
llama-cpp-python>=0.2.13

[openai]
openai>=0.26.0
tiktoken>=0.2.0
