requirements.txt

# Comment to bump caches
coloredlogs==14.0
datasets<2.2.2,>=1.4.1
docutils<0.16,>=0.14
emoji==0.5.4
fairscale~=0.4.1
docformatter~=1.4.0
flake8-bugbear==19.8.0
flake8==3.7.8
fuzzywuzzy==0.18.0
google-cloud-storage==1.43.0
iopath~=0.1.8
gitdb2==2.0.5
GitPython==3.1.37
hydra-core>=1.1.0
ipython==7.31.1
torch==2.0.0
torchvision==0.15.1
joblib==1.2.0
nltk==3.6.6
omegaconf>=2.1.1
pandas==1.4.0
pytest_regressions==2.4.2
pytest==7.3.0
pexpect==4.7.0
Pillow==10.0.1
py-gfm==1.0.2
py-rouge==1.1
pyyaml==5.4
pyzmq==18.1.0
regex>=2021.8.3
myst-parser<1
attrs~=20.2.0
requests-mock==1.7.0
requests<3,>=2.21.0
# scikit-learn==0.23.1
scipy==1.10.0
sh==1.12.14
sphinx_rtd_theme==0.4.3
sphinx-autodoc-typehints~=1.10.3
Sphinx~=5.1.0
subword-nmt==0.3.7
tensorboardX<=2.5.0
tokenizers>=0.13.3
tomli>=2.0.0
torchtext==0.15.1
tornado==6.3.2
tqdm~=4.62.1
typing-extensions==4.5.0
Unidecode==1.1.1
urllib3<1.27,>=1.26.5
websocket-client==0.56.0
jsonlines==1.2.0
numpy~=1.23.0
markdown<=3.3.2 # Pin to something that works so tests are happy
jinja2==3.0.3
ninja~=1.10.2.3
protobuf<=3.20.3, >=3.8.0
contractions~=0.1.72
fsspec~=2022.2.0
google-api-core<=2.11.0
openai<=0.27.7
litellm>=0.1.400