Skip to content

Commit be9b0e8

Browse files
authored
[CI] add back sacrebleu (and document why) (#37700)
* example test * add back dep * dev-ci * dev-ci
1 parent 1d7d7a9 commit be9b0e8

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

setup.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,10 @@
163163
"rjieba",
164164
"rouge-score!=0.0.7,!=0.0.8,!=0.1,!=0.1.1",
165165
"ruff==0.11.2",
166+
# `sacrebleu` not used in `transformers`. However, it is needed in several tests, when a test calls
167+
# `evaluate.load("sacrebleu")`. This metric is used in the examples that we use to test the `Trainer` with, in the
168+
# `Trainer` tests (see references to `run_translation.py`).
169+
"sacrebleu>=1.4.12,<2.0.0",
166170
"sacremoses",
167171
"safetensors>=0.4.3",
168172
"sagemaker>=2.31.0",
@@ -352,6 +356,7 @@ def run(self):
352356
"tensorboard",
353357
"pydantic",
354358
"sentencepiece",
359+
"sacrebleu", # needed in trainer tests, see references to `run_translation.py`
355360
)
356361
+ extras["retrieval"]
357362
+ extras["modelcreation"]

src/transformers/dependency_versions_table.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@
6969
"rjieba": "rjieba",
7070
"rouge-score": "rouge-score!=0.0.7,!=0.0.8,!=0.1,!=0.1.1",
7171
"ruff": "ruff==0.11.2",
72+
"sacrebleu": "sacrebleu>=1.4.12,<2.0.0",
7273
"sacremoses": "sacremoses",
7374
"safetensors": "safetensors>=0.4.3",
7475
"sagemaker": "sagemaker>=2.31.0",

0 commit comments

Comments
 (0)