From de7236bd5d53991f0e631a5868ffd3d0eae8cd8e Mon Sep 17 00:00:00 2001
From: Simon Meierhans <simonmeierhans@google.com>
Date: Fri, 11 Sep 2026 07:10:01 -0700
Subject: [PATCH] Fix contention timeout.

PiperOrigin-RevId: 979802322
---
 dgf/src/gbbs/BUILD | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/dgf/src/gbbs/BUILD b/dgf/src/gbbs/BUILD
index 6562d0f..b905d17 100644
--- a/dgf/src/gbbs/BUILD
+++ b/dgf/src/gbbs/BUILD
@@ -91,6 +91,11 @@ py_library(
 py_test(
     name = "connected_components_test",
     srcs = ["connected_components_test.py"],
+    # Parlay sizes its thread pool from std::thread::hardware_concurrency(),
+    # which reports the machine's core count rather than the CPUs this test is
+    # actually allocated. The resulting oversubscription makes the test time
+    # out, so pin the pool size here.
+    env = {"PARLAY_NUM_THREADS": "1"},
     deps = [
         ":connected_components_py",
         ":loader",
@@ -108,6 +113,8 @@ py_library(
 py_test(
     name = "loader_test",
     srcs = ["loader_test.py"],
+    # See the note on :connected_components_test above.
+    env = {"PARLAY_NUM_THREADS": "1"},
     deps = [
         ":loader",
         # absl/testing:absltest dep,