Skip to content

Commit fd04713

Browse files
committed
fix(scheduler): add capacity report compatibility shim
1 parent 224a00d commit fd04713

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

src/scheduling/scheduler.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,13 @@ def has_full_pipeline(self) -> bool:
144144
"""Check if there is a full pipeline among ACTIVE nodes."""
145145
return self.node_manager.has_full_pipeline(self.num_layers)
146146

147+
def report_pipeline_capacity(
148+
self,
149+
ready_only: bool = True,
150+
) -> Tuple[Optional[Dict[int, Tuple[int, int]]], int, int]:
151+
"""Backward-compatible capacity report delegated to ``NodeManager``."""
152+
return self.node_manager.report_pipeline_capacity(ready_only=ready_only)
153+
147154
def bootstrap(self, reboot: bool = False) -> bool:
148155
"""Initial Node Allocation Assignment."""
149156
logger.info("[Scheduler] Starting Bootstrap")

0 commit comments

Comments
 (0)