diff --git a/tests/crowdsourcing/tasks/model_chat/test_model_chat.py b/tests/crowdsourcing/tasks/model_chat/test_model_chat.py index 4970b94b1ba..24200ffb40a 100644 --- a/tests/crowdsourcing/tasks/model_chat/test_model_chat.py +++ b/tests/crowdsourcing/tasks/model_chat/test_model_chat.py @@ -82,6 +82,7 @@ def setUp(self) -> None: def tearDown(self) -> None: self._teardown() + @testing_utils.retry(ntries=3) def test_base_task(self): with testing_utils.tempdir() as tmpdir: diff --git a/tests/crowdsourcing/tasks/model_chat/test_model_image_chat.py b/tests/crowdsourcing/tasks/model_chat/test_model_image_chat.py index ba023b3238a..de8446ddb8d 100644 --- a/tests/crowdsourcing/tasks/model_chat/test_model_image_chat.py +++ b/tests/crowdsourcing/tasks/model_chat/test_model_image_chat.py @@ -64,6 +64,7 @@ def setUp(self) -> None: def tearDown(self) -> None: self._teardown() + @testing_utils.retry(ntries=3) def test_base_task(self): with testing_utils.tempdir() as tmpdir: diff --git a/tests/crowdsourcing/tasks/test_chat_demo.py b/tests/crowdsourcing/tasks/test_chat_demo.py index f58cebfa443..8522880280c 100644 --- a/tests/crowdsourcing/tasks/test_chat_demo.py +++ b/tests/crowdsourcing/tasks/test_chat_demo.py @@ -10,6 +10,8 @@ import os import unittest +import parlai.utils.testing as testing_utils + # Desired inputs/outputs EXPECTED_STATE_AGENT_0 = { @@ -372,6 +374,7 @@ def setUp(self) -> None: def tearDown(self) -> None: self._teardown() + @testing_utils.retry(ntries=3) def test_base_task(self): # # Setup