Only starting measure worker in new process

google · Aug 22, 2024 · a365f7e · a365f7e
1 parent 05d3668
commit a365f7e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/experiment/measurer/measure_manager.py b/experiment/measurer/measure_manager.py
@@ -959,9 +959,13 @@ def start_workers(self, request_queue, response_queue, pool):
             worker_subscription_path, worker_request_queue_topic_path)
 
         for _ in range(self.measurers_cpus):
-            google_cloud_worker = measure_worker.GoogleCloudMeasureWorker(
-                config)
-            pool.apply_async(google_cloud_worker.measure_worker_loop)
+
+            def start_measure_workers_and_start_loop():
+                google_cloud_worker = measure_worker.GoogleCloudMeasureWorker(
+                    config)
+                google_cloud_worker.measure_worker_loop()
+
+            pool.apply_async(start_measure_workers_and_start_loop)
 
     def get_result_from_response_queue(self, response_queue: str):
 

diff --git a/experiment/measurer/test_measure_manager.py b/experiment/measurer/test_measure_manager.py
@@ -672,5 +672,3 @@ def test_gcloud_measure_manager_start_workers(_mock_subscriber, _mock_publisher,
         gcloud_measure_manager.start_workers('request-queue-topic',
                                              'response-queue-topic', pool)
         assert pool.apply_async.call_count == cpus_available
-        pool.apply_async.assert_called_with(
-            mock_gcloud_measure_worker().measure_worker_loop)