Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 2 additions & 4 deletions clients/python/src/taskbroker_client/worker/workerchild.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
import contextlib
import logging
import multiprocessing
import os
import queue
import signal
import threading
Expand Down Expand Up @@ -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()
)
Expand Down Expand Up @@ -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,
Expand Down
Loading