From d9bf753fb99764dac19a0829ae9d69f23eb96d7b Mon Sep 17 00:00:00 2001 From: Tom Aarsen <37621491+tomaarsen@users.noreply.github.com> Date: Mon, 13 Jan 2025 11:24:27 +0100 Subject: [PATCH] [`tests`] Add 'save_strategy="no"' in tests to counteract transformers v4.48.0 bug (#582) --- tests/span/test_model_card.py | 1 + tests/test_model_card.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tests/span/test_model_card.py b/tests/span/test_model_card.py index 60fd830b..6e4585f8 100644 --- a/tests/span/test_model_card.py +++ b/tests/span/test_model_card.py @@ -26,6 +26,7 @@ def test_model_card(absa_dataset: Dataset, tmp_path: Path) -> None: logging_steps=1, max_steps=2, eval_strategy="steps", + save_strategy="no", ) trainer = AbsaTrainer( model=model, diff --git a/tests/test_model_card.py b/tests/test_model_card.py index 910f7237..9bdb199c 100644 --- a/tests/test_model_card.py +++ b/tests/test_model_card.py @@ -36,6 +36,7 @@ def test_model_card(tmp_path: Path) -> None: logging_steps=1, max_steps=2, eval_strategy="steps", + save_strategy="no", ) trainer = Trainer( model=model,