2023-07-04 02:31:55 +08:00
|
|
|
# formatting
|
|
|
|
|
yapf==0.32.0
|
2023-12-14 16:19:41 +08:00
|
|
|
toml==0.10.2
|
2024-02-22 10:56:01 +08:00
|
|
|
tomli==2.0.1
|
2023-11-21 03:58:01 +08:00
|
|
|
ruff==0.1.5
|
2024-02-22 10:56:01 +08:00
|
|
|
codespell==2.2.6
|
2024-03-25 22:59:47 +08:00
|
|
|
isort==5.13.2
|
2023-07-04 02:31:55 +08:00
|
|
|
|
|
|
|
|
# type checking
|
|
|
|
|
mypy==0.991
|
|
|
|
|
types-PyYAML
|
|
|
|
|
types-requests
|
|
|
|
|
types-setuptools
|
|
|
|
|
|
|
|
|
|
# testing
|
2023-05-24 08:58:51 +08:00
|
|
|
pytest
|
2023-09-01 10:19:43 +08:00
|
|
|
pytest-forked
|
2023-09-17 15:29:08 +08:00
|
|
|
pytest-asyncio
|
2024-03-02 04:47:51 +08:00
|
|
|
pytest-rerunfailures
|
2024-03-18 05:56:30 +08:00
|
|
|
pytest-shard
|
2024-01-15 04:37:58 +08:00
|
|
|
httpx
|
|
|
|
|
einops # required for MPT
|
2024-01-17 13:33:14 +08:00
|
|
|
openai
|
|
|
|
|
requests
|
2024-02-23 01:58:29 +08:00
|
|
|
ray
|
2024-03-10 09:14:16 +08:00
|
|
|
peft
|
2024-03-26 05:16:30 +08:00
|
|
|
awscli
|
2024-03-29 12:06:40 +08:00
|
|
|
ai2-olmo # required for OLMo
|
2024-03-07 16:39:28 +08:00
|
|
|
|
|
|
|
|
# Benchmarking
|
|
|
|
|
aiohttp
|
2024-03-26 05:16:30 +08:00
|
|
|
|
|
|
|
|
# Multimodal
|
|
|
|
|
pillow
|