Update ruff to 0.11.2 (#36962)
* update * update * update --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -180,9 +180,9 @@ class Seq2seqTrainerTester(TestCasePlus):
|
||||
for num_return_sequences in range(3, 0, -1):
|
||||
gen_config.num_return_sequences = num_return_sequences
|
||||
metrics = trainer.evaluate(eval_dataset=prepared_dataset, generation_config=gen_config)
|
||||
assert (
|
||||
metrics["eval_samples"] == dataset_len * num_return_sequences
|
||||
), f"Got {metrics['eval_samples']}, expected: {dataset_len * num_return_sequences}"
|
||||
assert metrics["eval_samples"] == dataset_len * num_return_sequences, (
|
||||
f"Got {metrics['eval_samples']}, expected: {dataset_len * num_return_sequences}"
|
||||
)
|
||||
|
||||
@require_torch
|
||||
def test_bad_generation_config_fail_early(self):
|
||||
|
||||
Reference in New Issue
Block a user