Fix fbcode//pearl:test_pearl - test_serialization (pearl.test.unit.with_pytorch.test_serialization.TestSerialization) (T278585348)

Differential Revision: D111401279

fbshipit-source-id: 4b6a9b665e2d3901ba763c0470cac623074d3099

Author: generatedunixname2129521104496973

test/unit/with_pytorch/test_serialization.py

@@ -23,6 +23,10 @@ def save_and_load_state_dict(origin: nn.Module, destination: nn.Module) -> None:
 
 class TestSerialization(TestCase):
     def test_serialization(self) -> None:
+        # Concurrent stress-run copies each default to one compute thread per
+        # core, oversubscribing the host and timing out; the models are tiny.
+        torch.set_num_threads(1)
+
         test_agent = TestAgentWithPyTorch()
         for (
             agent_type,