Skip to content

Commit af98cfa

Browse files
chore: pre-commit refactor
1 parent 18bba89 commit af98cfa

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

src/forge/actors/generator.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,9 @@ def split_keys(keys):
287287
return state_dict
288288

289289
@endpoint
290-
async def generate(self, prompt: str, *, priority: int = 0, n: int | None = None) -> list[Completion]:
290+
async def generate(
291+
self, prompt: str, *, priority: int = 0, n: int | None = None
292+
) -> list[Completion]:
291293
"""Generate a response for the given prompt
292294
293295
Args:

src/forge/controller/launcher.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,6 @@
1717
import monarch
1818

1919
import torchx.specs as specs
20-
21-
from forge.types import Launcher, LauncherConfig
2220
from monarch._rust_bindings.monarch_hyperactor.alloc import AllocConstraints
2321
from monarch._rust_bindings.monarch_hyperactor.channel import ChannelTransport
2422

@@ -29,6 +27,8 @@
2927
from monarch.tools.commands import create, info
3028
from monarch.tools.config import Config, Workspace
3129

30+
from forge.types import Launcher, LauncherConfig
31+
3232
_MAST_AVAILABLE = False
3333

3434
try:

0 commit comments

Comments
 (0)