create tasks

Author: casteryh

src/forge/actors/generator.py

@@ -294,10 +294,13 @@ def split_keys(keys):
 
         futures = []
         for i, names in enumerate(split_keys(hf_param_names)):
-            fut = self.weight_fetchers.slice(procs=i).fetch.call_one(
-                version=version, param_names=names
-            )
-            futures.append(fut)
+
+            async def _fetch():
+                await self.weight_fetchers.slice(procs=i).fetch.call_one(
+                    version=version, param_names=names
+                )
+
+            futures.append(asyncio.create_task(_fetch()))
 
         sub_state_dicts = [await fut for fut in futures]