From 64a10e86efda76470189e92f68a93d4c1a040e4e Mon Sep 17 00:00:00 2001 From: Tom Aarsen Date: Fri, 10 Jan 2025 16:45:00 +0100 Subject: [PATCH] Add 'save_strategy="no"' in tests to counteract transformers v4.48.0 bug --- 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,