cattrs~=22.2.0
dacite~=1.6.0
importlib-resources~=5.10.0
Mako~=1.2.3
numpy~=1.23.3
pyhocon~=0.3.59
retrying~=1.3.4
spacy~=3.5.3
tqdm~=4.64.1
zstandard~=0.18.0
sqlitedict~=1.7.0
bottle~=0.12.23
pymongo~=4.2.0
datasets~=2.5.2
pyarrow>=11.0.0
nltk~=3.7
pyext~=0.7
rouge-score~=0.1.2
scipy~=1.10.0
uncertainty-calibration~=0.1.4
scikit-learn~=1.1.2
transformers~=4.33.1
torch<3.0.0,>=1.12.1
torchvision<3.0.0,>=0.13.1
google-api-python-client~=2.64.0

[aleph-alpha]
aleph-alpha-client~=2.14.0
tokenizers~=0.13.3

[all]
crfm-helm[proxy-server]
crfm-helm[human-evaluation]
crfm-helm[scenarios]
crfm-helm[metrics]
crfm-helm[plots]
crfm-helm[slurm]
crfm-helm[cleva]
crfm-helm[images]
crfm-helm[models]

[anthropic]
anthropic~=0.2.5
websocket-client~=1.3.2

[cleva]
unidecode==1.3.6
pypinyin==0.49.0
jieba==0.42.1
opencc==1.1.6
langdetect==1.0.9

[dev]
pytest~=7.2.0
black~=22.10.0
mypy~=0.982
pre-commit~=2.20.0
flake8~=5.0.4

[human-evaluation]
scaleapi~=2.13.0
surge-api~=1.1.0

[images]
accelerate~=0.23.0
pillow~=9.4.0

[metrics]
numba~=0.56.4
pytrec_eval==0.5
sacrebleu~=2.2.1
summ-eval~=0.892

[models]
crfm-helm[aleph-alpha]
crfm-helm[anthropic]
crfm-helm[openai]
crfm-helm[tsinghua]
crfm-helm[yandex]

[openai]
openai~=0.27.8
tiktoken~=0.3.3

[plots]
colorcet~=3.0.1
matplotlib~=3.6.0
seaborn~=0.11.0

[proxy-server]
gunicorn~=20.1.0

[scenarios]
gdown~=4.4.0
sympy~=1.11.1
xlrd~=2.0.1

[slurm]
simple-slurm~=0.2.6

[tsinghua]
icetk~=0.0.4

[yandex]
sentencepiece~=0.1.97
