Fix fbcode//pearl:test_pearl - test_construction_of_all_types (pearl.test.unit.with_pytorch.test_agent.TestAgentWithPyTorch) (T283234158)

Differential Revision: D114591826

fbshipit-source-id: d1aa76d4e827f229608f58e53d6cc06e7b4037ea

Author: generatedunixname2129521104496973

test/unit/with_pytorch/test_agent.py

@@ -1196,6 +1196,10 @@ def get_trained_qr_dqn_agent_and_environment(
     """
 
     def test_construction_of_all_types(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)
+
         # The offline agents (IQL and CQL) fetch their training datasets from an
         # external URL, unreachable under network isolation (e.g. network_access=none).
         # Detect isolation via the local-forkserver env var or the RE marker file,