Skip to content
This repository was archived by the owner on Dec 16, 2022. It is now read-only.

Commit 2e7acb0

Browse files
authored
Mark QaNet test as flaky (#2481)
* Mark QaNet test as flaky * Import the decorator 🤦
1 parent 23ea590 commit 2e7acb0

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

allennlp/tests/models/reading_comprehension/qanet_test.py

+3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
# pylint: disable=no-self-use,invalid-name
2+
from flaky import flaky
23
import numpy
34
from numpy.testing import assert_almost_equal
5+
46
from allennlp.common import Params
57
from allennlp.data import DatasetReader, Vocabulary
68
from allennlp.common.testing import ModelTestCase
@@ -39,6 +41,7 @@ def test_forward_pass_runs_correctly(self):
3941
assert span_end < self.instances[0].fields['passage'].sequence_length()
4042
assert isinstance(output_dict['best_span_str'][0], str)
4143

44+
@flaky
4245
def test_model_can_train_save_and_load(self):
4346
self.ensure_model_can_train_save_and_load(self.param_file, tolerance=1e-4)
4447

0 commit comments

Comments
 (0)