Skip to content

Commit c752c64

Browse files
committed
run lint
1 parent 9138c61 commit c752c64

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

src/sglang_diffusion_routing/router/diffusion_router.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(self, args, verbose: bool = False):
3838
self.worker_video_support: dict[str, bool | None] = {}
3939
# quarantined workers excluded from routing
4040
self.dead_workers: set[str] = set()
41-
# record workers in sleeping status
41+
# record workers in sleeping status
4242
self.sleeping_workers: set[str] = set()
4343
# video_id -> worker URL mapping for stable query routing
4444
self.video_job_to_worker: dict[str, str] = {}
@@ -724,15 +724,16 @@ async def release_memory_occupation(self, request: Request):
724724
headers.pop("content-length", None)
725725
headers.setdefault("content-type", "application/json")
726726

727-
results = await self._broadcast_to_workers("release_memory_occupation", body, headers)
727+
results = await self._broadcast_to_workers(
728+
"release_memory_occupation", body, headers
729+
)
728730

729731
for item in results:
730732
if item.get("status_code") == 200:
731733
self.sleeping_workers.add(item["worker_url"])
732734

733735
return JSONResponse(content={"results": results})
734736

735-
736737
async def resume_memory_occupation(self, request: Request):
737738
"""Broadcast wake to all healthy workers and unmark sleeping on success."""
738739
healthy_workers = [
@@ -748,7 +749,9 @@ async def resume_memory_occupation(self, request: Request):
748749
headers.pop("content-length", None)
749750
headers.setdefault("content-type", "application/json")
750751

751-
results = await self._broadcast_to_workers("resume_memory_occupation", body, headers)
752+
results = await self._broadcast_to_workers(
753+
"resume_memory_occupation", body, headers
754+
)
752755

753756
for item in results:
754757
if item.get("status_code") == 200:

0 commit comments

Comments
 (0)