Set multiprocessing in test case with error handling.

allenai · Feb 28, 2025 · 41e42a6 · 41e42a6
1 parent 056bc34
commit 41e42a6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/python/test_code_composition.py b/tests/python/test_code_composition.py
@@ -40,7 +40,11 @@ def baz():
 
 class TestDolmaCodeProseCompositionClassifier(TestCase):
     def setUp(self) -> None:
-        multiprocessing.set_start_method("spawn")
+        try:
+            multiprocessing.set_start_method("spawn")
+        except Exception:
+            pass
+
         self.code_composition_tagger = CodeProseCompositionClassifier()
 
     def test_prose_text(self):