Rebuild ScheduledRequestInfo to recognize MeasuredRequestTimings schema change

sjmonson · sjmonson · commit a3bc82437f8c · 2025-09-02T20:50:15.000-04:00
diff --git a/src/guidellm/backend/objects.py b/src/guidellm/backend/objects.py
@@ -13,6 +13,7 @@
 
 from guidellm.scheduler import (
     MeasuredRequestTimings,
+    ScheduledRequestInfo,
     SchedulerMessagingPydanticRegistry,
 )
 from guidellm.utils import StandardBaseModel
@@ -155,4 +156,7 @@ class GenerationRequestTimings(MeasuredRequestTimings):
     )
 
 
+# Rebuild ScheduledRequestInfo to recognize MeasuredRequestTimings schema change
+ScheduledRequestInfo.model_rebuild(force=True)
+
 SchedulerMessagingPydanticRegistry.register_decorator(GenerationRequestTimings)

Original file line number	Diff line number	Diff line change
`@@ -13,6 +13,7 @@`
`13`	`13`
`14`	`14`	`from guidellm.scheduler import (`
`15`	`15`	`MeasuredRequestTimings,`
	`16`	`+ ScheduledRequestInfo,`
`16`	`17`	`SchedulerMessagingPydanticRegistry,`
`17`	`18`	`)`
`18`	`19`	`from guidellm.utils import StandardBaseModel`
`@@ -155,4 +156,7 @@ class GenerationRequestTimings(MeasuredRequestTimings):`
`155`	`156`	`)`
`156`	`157`
`157`	`158`
	`159`	`+# Rebuild ScheduledRequestInfo to recognize MeasuredRequestTimings schema change`
	`160`	`+ScheduledRequestInfo.model_rebuild(force=True)`
	`161`	`+`
`158`	`162`	`SchedulerMessagingPydanticRegistry.register_decorator(GenerationRequestTimings)`