Skip to content

Commit 89b145d

Browse files
committed
Add info property to sat constraint
Signed-off-by: Samuel Monson <[email protected]>
1 parent 1d16296 commit 89b145d

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

src/guidellm/scheduler/constraints/saturation.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -334,6 +334,25 @@ def __init__(
334334
self.enabled = enabled
335335
self.reset()
336336

337+
@property
338+
def info(self) -> dict[str, Any]:
339+
"""
340+
Get current constraint configuration and state information.
341+
:return: Dictionary containing configuration parameters.
342+
"""
343+
344+
return {
345+
"type_": "over_saturation",
346+
"minimum_duration": self.minimum_duration,
347+
"minimum_ttft": self.minimum_ttft,
348+
"maximum_window_seconds": self.maximum_window_seconds,
349+
"maximum_window_ratio": self.maximum_window_ratio,
350+
"minimum_window_size": self.minimum_window_size,
351+
"moe_threshold": self.moe_threshold,
352+
"confidence": self.confidence,
353+
"enabled": self.enabled,
354+
}
355+
337356
def reset(self) -> None:
338357
"""
339358
Reset all internal state to initial values.

0 commit comments

Comments
 (0)