Add 'save_strategy="no"' in tests to counteract transformers v4.48.0 bug

huggingface · Jan 13, 2025 · 64a10e8 · 64a10e8
1 parent 301603b
commit 64a10e8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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,