fix fetcher call

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

Original file line number	Diff line number	Diff line change
`@@ -294,8 +294,8 @@ 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_weights.call_one(`
`298`		`- names`
	`297`	`+ fut = await self.weight_fetchers.slice(procs=i).fetch.call_one(`
	`298`	`+ version, names`
`299`	`299`	`)`
`300`	`300`	`futures.append(fut)`
`301`	`301`