typo

casteryh · casteryh · commit 2798f2e6b78b · 2025-10-14T17:19:26.000-07:00
diff --git a/src/forge/actors/generator.py b/src/forge/actors/generator.py
@@ -407,7 +407,7 @@ async def update_weights(self, version: int) -> None:
         # TODO: move this logic to Generator once we can make sure Generator and GeneratorWorker are on the same host.
         if not self.use_dcp:
             fetch_task = asyncio.create_task(
-                self.policy_worker._fetch_weights.choose(version)
+                self.generator_worker._fetch_weights.choose(version)
             )
         # Serialize updates (only one update at a time)
         async with self.update_lock:

Original file line number	Diff line number	Diff line change
`@@ -407,7 +407,7 @@ async def update_weights(self, version: int) -> None:`
`407`	`407`	`# TODO: move this logic to Generator once we can make sure Generator and GeneratorWorker are on the same host.`
`408`	`408`	`if not self.use_dcp:`
`409`	`409`	`fetch_task = asyncio.create_task(`
`410`		`- self.policy_worker._fetch_weights.choose(version)`
	`410`	`+ self.generator_worker._fetch_weights.choose(version)`
`411`	`411`	`)`
`412`	`412`	`# Serialize updates (only one update at a time)`
`413`	`413`	`async with self.update_lock:`