format

Author: JenniferWang

apps/toy_rl/sumdigits.py

@@ -397,13 +397,7 @@ async def main(cfg: DictConfig):
 
     # ---- Setup services ---- #
     await ts.initialize()
-    (
-        dataloader,
-        policy,
-        trainer,
-        replay_buffer,
-        reward_actor,
-    ) = await asyncio.gather(
+    (dataloader, policy, trainer, replay_buffer, reward_actor,) = await asyncio.gather(
         DatasetActor.options(**cfg.services.dataset).as_service(**cfg.dataset),
         Policy.options(**cfg.services.policy).as_service(**cfg.policy),
         Trainer.options(**cfg.services.trainer).as_service(**cfg.trainer),

src/forge/actors/replay_buffer.py

@@ -8,10 +8,10 @@
 from dataclasses import dataclass
 from typing import Any, Callable
 
-from forge.controller import ForgeActor
-
 from monarch.actor import endpoint
 
+from forge.controller import ForgeActor
+
 
 @dataclass
 class ReplayBuffer(ForgeActor):