Skip to content

GitLab

Unverified Commit 31ec424b authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Add decorator to flaky test (#19674)

parent a929f81e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/pytorch/test_accelerate_examples.py
View file @ 31ec424b
...@@ -26,7 +26,7 @@ from unittest import mock ...@@ -26,7 +26,7 @@ from unittest import mock
import torch import torch
from accelerate.utils import write_basic_config from accelerate.utils import write_basic_config
from transformers.testing_utils import TestCasePlus, get_gpu_count, run_command, slow, torch_device from transformers.testing_utils import TestCasePlus, get_gpu_count, is_flaky, run_command, slow, torch_device
from transformers.utils import is_apex_available from transformers.utils import is_apex_available
...@@ -176,6 +176,7 @@ class ExamplesTestsNoTrainer(TestCasePlus): ...@@ -176,6 +176,7 @@ class ExamplesTestsNoTrainer(TestCasePlus):
self.assertTrue(os.path.exists(os.path.join(tmp_dir, "epoch_0"))) self.assertTrue(os.path.exists(os.path.join(tmp_dir, "epoch_0")))
self.assertTrue(os.path.exists(os.path.join(tmp_dir, "ner_no_trainer"))) self.assertTrue(os.path.exists(os.path.join(tmp_dir, "ner_no_trainer")))
@is_flaky()
@mock.patch.dict(os.environ, {"WANDB_MODE": "offline"}) @mock.patch.dict(os.environ, {"WANDB_MODE": "offline"})
def test_run_squad_no_trainer(self): def test_run_squad_no_trainer(self):
tmp_dir = self.get_auto_remove_tmp_dir() tmp_dir = self.get_auto_remove_tmp_dir()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment