Skip to content

Commit 22cca14

Browse files
test: over-saturation stopping
Add unit tests for over-saturation detection and constraint functionality. Signed-off-by: Alon Kellner <[email protected]>
1 parent e4a66ec commit 22cca14

File tree

2 files changed

+625
-1
lines changed

2 files changed

+625
-1
lines changed

src/guidellm/scheduler/constraints/over_saturation.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -437,6 +437,8 @@ def validated_kwargs(cls, stop_over_saturated: bool, **kwargs) -> dict[str, Any]
437437
"""
438438
aliases = ["stop_over_saturated", "stop_over_sat", "stop_osd"]
439439
for alias in aliases:
440-
stop_over_saturated = stop_over_saturated or kwargs.get(alias)
440+
alias_value = kwargs.get(alias)
441+
if alias_value is not None:
442+
stop_over_saturated = stop_over_saturated or alias_value
441443

442444
return {"stop_over_saturated": stop_over_saturated}

0 commit comments

Comments
 (0)