Skip to content
This repository has been archived by the owner on Oct 11, 2024. It is now read-only.

Commit

Permalink
Seperate dev requirements into lint and test (vllm-project#5474)
Browse files Browse the repository at this point in the history
  • Loading branch information
Yard1 authored and robertgshaw2-neuralmagic committed Jun 16, 2024
1 parent 2752570 commit fa895d1
Show file tree
Hide file tree
Showing 4 changed files with 42 additions and 40 deletions.
2 changes: 2 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
pip install -r requirements-cuda.txt

# install development dependencies
COPY requirements-lint.txt requirements-lint.txt
COPY requirements-test.txt requirements-test.txt
COPY requirements-dev.txt requirements-dev.txt
RUN --mount=type=cache,target=/root/.cache/pip \
pip install -r requirements-dev.txt
Expand Down
44 changes: 4 additions & 40 deletions requirements-dev.txt
Original file line number Diff line number Diff line change
@@ -1,41 +1,5 @@
# formatting
yapf==0.32.0
toml==0.10.2
tomli==2.0.1
ruff==0.1.5
codespell==2.3.0
isort==5.13.2
clang-format==18.1.5
-r requirements-lint.txt
-r requirements-test.txt

# type checking
mypy==1.9.0
types-PyYAML
types-requests==2.31.0.2
types-setuptools

# testing
autoawq
datasets
pytest
tensorizer>=2.9.0
pytest-forked
pytest-asyncio
pytest-rerunfailures
pytest-shard

# testing utils
awscli
einops # required for MPT
httpx
peft
requests==2.31
ray
sentence-transformers # required for embedding
optimum # required for hf gptq baselines
auto-gptq # required for hf gptq baselines

# Benchmarking
aiohttp

# quantization
bitsandbytes==0.42.0
# Avoid adding requirements directly to this file.
# Instead, modify the two files referenced above.
14 changes: 14 additions & 0 deletions requirements-lint.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
# formatting
yapf==0.32.0
toml==0.10.2
tomli==2.0.1
ruff==0.1.5
codespell==2.3.0
isort==5.13.2
clang-format==18.1.5

# type checking
mypy==1.9.0
types-PyYAML
types-requests
types-setuptools
22 changes: 22 additions & 0 deletions requirements-test.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
# testing
pytest
tensorizer>=2.9.0
pytest-forked
pytest-asyncio
pytest-rerunfailures
pytest-shard

# testing utils
awscli
einops # required for MPT
httpx
peft
requests
ray
sentence-transformers # required for embedding

# Benchmarking
aiohttp

# quantization
bitsandbytes==0.42.0

0 comments on commit fa895d1

Please sign in to comment.