Retry flaky crowdsourcing tests

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:

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:

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