Skip to content

Commit

Permalink
Tidy up dependency groups
Browse files Browse the repository at this point in the history
  • Loading branch information
lewtun committed Mar 2, 2024
1 parent b9d0277 commit 00716fa
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 10 deletions.
5 changes: 1 addition & 4 deletions .github/workflows/quality.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -24,10 +24,7 @@ jobs:
- name: Install dependencies
run: |
python -m pip install --upgrade pip
python setup.py egg_info
sed '/^$/q' src/lighteval.egg-info/requires.txt > src/lighteval.egg-info/requires_lite.txt
python -m pip install ruff -c src/lighteval.egg-info/requires_lite.txt
rm -rf src/lighteval.egg-info
python -m pip install ".[quality]"
- name: Code quality
run: |
make quality
2 changes: 1 addition & 1 deletion .github/workflows/tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ jobs:
cache: 'pip'
- name: Install lighteval in editable mode
run: |
pip install -e .[accelerate]
pip install -e .[dev]
- name: Get cached files
uses: actions/cache@v2
id: get-cache
Expand Down
9 changes: 4 additions & 5 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -50,12 +50,10 @@ keywords = ["evaluation", "nlp", "llm"]
dependencies = [
# Base dependencies
"transformers>=4.38.0",
"huggingface_hub==0.20.3",
"huggingface_hub>=0.20.3",
"torch>=2.0",
"GitPython==3.1.31", # for logging
"datasets>=2.14.0",
# Test
"pytest==7.4.0",
# Prettiness
"termcolor==2.3.0",
"pytablewriter",
Expand All @@ -72,8 +70,6 @@ dependencies = [
"sentencepiece>=0.1.99",
"protobuf==3.20.*", # pinned for sentencepiece compat
"pycountry",
# Code style
"ruff==v0.2.2",
]

[project.optional-dependencies]
Expand All @@ -87,6 +83,9 @@ nanotron = [
"nanotron",
"tensorboardX"
]
quality = ["ruff==v0.2.2"]
tests = ["pytest==7.4.0"]
dev = ["lighteval[accelerate,quality,tests]"]


[project.urls]
Expand Down

0 comments on commit 00716fa

Please sign in to comment.