From f66f8ae9000f94bdc329eb34266c0976b94e0c8a Mon Sep 17 00:00:00 2001
From: Ben McKerry <110857332+bmckerry@users.noreply.github.com>
Date: Tue, 7 Jul 2026 15:19:41 -0400
Subject: [PATCH] ref(workerchild): don't set future tracking env var

---
 clients/python/src/taskbroker_client/worker/workerchild.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/clients/python/src/taskbroker_client/worker/workerchild.py b/clients/python/src/taskbroker_client/worker/workerchild.py
index 07363d6b..d8f206e1 100644
--- a/clients/python/src/taskbroker_client/worker/workerchild.py
+++ b/clients/python/src/taskbroker_client/worker/workerchild.py
@@ -3,7 +3,6 @@
 import contextlib
 import logging
 import multiprocessing
-import os
 import queue
 import signal
 import threading
@@ -546,6 +545,8 @@ def check_task_future_completion(
             clear_current_task()
             processed_task_count += 1
 
+            # To have Taskworker track futures, set the env var `ARROYO_TRACK_PRODUCER_FUTURES = True`
+            # in the worker process
             task_produced_futures = (
                 TaskProducer.collect_futures() | FutureTrackingProducer.collect_futures()
             )
@@ -854,9 +855,6 @@ def _task_execution_complete(
     # Tell the parent that this child has warmed up and is ready to consume tasks
     messages.put_nowait(ChildMessage(child_id, "running"))
 
-    # Tell FutureTrackingProducer to track producer futures in this process
-    os.environ["ARROYO_TRACK_PRODUCER_FUTURES"] = "True"
-
     # Run the worker loop
     run_worker(
         child_tasks,