Skip to content

Commit ca5363c

Browse files
committed
Treat OverSaturationConstraint as a Constraint
Signed-off-by: Samuel Monson <[email protected]>
1 parent 3bd986b commit ca5363c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/guidellm/scheduler/constraints/saturation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ def check_slope(self, effective_n: float) -> bool:
269269
return (slope > 0) and (margin_of_error < self.moe_threshold)
270270

271271

272-
class OverSaturationConstraint: # type: ignore[misc]
272+
class OverSaturationConstraint(Constraint):
273273
"""
274274
Constraint that detects and stops execution when over-saturation is detected.
275275
@@ -632,7 +632,7 @@ def create_constraint(self, **_kwargs) -> Constraint:
632632
:param _kwargs: Additional keyword arguments (unused)
633633
:return: Configured OverSaturationConstraint instance ready for use
634634
"""
635-
return OverSaturationConstraint( # type: ignore[return-value]
635+
return OverSaturationConstraint(
636636
minimum_duration=self.min_seconds,
637637
minimum_ttft=self.minimum_ttft,
638638
maximum_window_seconds=self.max_window_seconds,

0 commit comments

Comments
 (0)