remove await

casteryh · casteryh · commit be85a94ae677 · 2025-10-15T13:11:22.000-07:00
diff --git a/src/forge/actors/generator.py b/src/forge/actors/generator.py
@@ -294,7 +294,7 @@ def split_keys(keys):
 
         futures = []
         for i, names in enumerate(split_keys(hf_param_names)):
-            fut = await self.weight_fetchers.slice(procs=i).fetch.call_one(
+            fut = self.weight_fetchers.slice(procs=i).fetch.call_one(
                 version=version, param_names=names
             )
             futures.append(fut)

Original file line number	Diff line number	Diff line change
`@@ -294,7 +294,7 @@ def split_keys(keys):`
`294`	`294`
`295`	`295`	`futures = []`
`296`	`296`	`for i, names in enumerate(split_keys(hf_param_names)):`
`297`		`- fut = await self.weight_fetchers.slice(procs=i).fetch.call_one(`
	`297`	`+ fut = self.weight_fetchers.slice(procs=i).fetch.call_one(`
`298`	`298`	`version=version, param_names=names`
`299`	`299`	`)`
`300`	`300`	`futures.append(fut)`