Skip to content

Commit 1eee287

Browse files
committed
Revert fetcher proc
1 parent 9740ba7 commit 1eee287

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

src/forge/actors/generator.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -239,12 +239,11 @@ def _spawn_fetchers(self):
239239
# TODO: this assumes the generator is on the same host as the worker
240240
# and only works for single host generators. Figure out how to support
241241
# generators with workers spanned across multiple hosts.
242-
pass
243-
# fetcher_procs = this_host().spawn_procs(
244-
# per_host={"procs": self.n_fetcher_procs}
245-
# )
246-
# self._fetcher_procs = fetcher_procs
247-
# self.weight_fetchers = fetcher_procs.spawn("weight_fetcher", _WeightFetcher)
242+
fetcher_procs = this_host().spawn_procs(
243+
per_host={"procs": self.n_fetcher_procs}
244+
)
245+
self._fetcher_procs = fetcher_procs
246+
self.weight_fetchers = fetcher_procs.spawn("weight_fetcher", _WeightFetcher)
248247

249248
def _start_processing(self):
250249
if self._run_task is None or self._run_task.done():

0 commit comments

Comments
 (0)