Fix most type errors for the IPC Queue

Signed-off-by: Jared O'Connell <[email protected]>

Author: jaredoconnell

src/guidellm/scheduler/constraints.py

@@ -1005,7 +1005,7 @@ def info(self) -> dict[str, Any]:
         return self.model_dump()
 
     def __call__(
-        self, state: SchedulerState, request: RequestInfo
+        self, state: SchedulerState, _request: RequestInfo
     ) -> SchedulerUpdateAction:
         create_exceeded = state.created_requests >= self.num_requests
         processed_exceeded = state.processed_requests >= self.num_requests

src/guidellm/scheduler/strategies.py

@@ -138,7 +138,9 @@ def init_processes_start(self, start_time: float):
                 "init_processes_timings"
             )
         if self._processes_start_time is None:
-            raise RuntimeError("_processes_lock is not None but _processes_start_time is None")
+            raise RuntimeError(
+                "_processes_lock is not None but _processes_start_time is None"
+            )
 
         with self._processes_lock:
             self._processes_start_time.value = start_time
@@ -156,7 +158,9 @@ async def get_processes_start_time(self) -> float:
                 "init_processes_timings"
             )
         if self._processes_start_time is None:
-            raise RuntimeError("_processes_lock is not None but _processes_start_time is None")
+            raise RuntimeError(
+                "_processes_lock is not None but _processes_start_time is None"
+            )
 
         while self._cached_processes_start_time is None:
             with self._processes_lock:
@@ -180,7 +184,9 @@ def next_request_index(self) -> int:
                 "init_processes_timings"
             )
         if self._processes_request_index is None:
-            raise RuntimeError("_processes_lock is not None but _processes_request_index is None")
+            raise RuntimeError(
+                "_processes_lock is not None but _processes_request_index is None"
+            )
 
         with self._processes_lock:
             self._processes_request_index.value += 1

src/guidellm/scheduler/worker.py

@@ -378,7 +378,9 @@ async def _process_next_request(self, target_start: float):
             # Process the request with the backend
             request_info.timings.resolve_start = time.time()
             self._send_update("in_progress", response, request, request_info)
-            async for resp, info in await self.backend.resolve(request, request_info, None):
+            async for resp, info in await self.backend.resolve(
+                request, request_info, None
+            ):
                 response = resp
                 request_info = info
 

src/guidellm/scheduler/worker_group.py

@@ -232,7 +232,7 @@ async def create_processes(self):
                     worker_index=rank,
                     max_buffer_send_size=None,
                     max_buffer_receive_size=per_proc_max_buffer_size,
-                ),
+                ),  # The non-group worker lacks the SchedulerState type. Type err.
                 backend=self.backend,
                 strategy=self.strategy,
                 async_limit=async_limit,
@@ -533,7 +533,7 @@ def requests_generator(
             self._locked_update(
                 info=None,
                 add_constraints={
-                    "requests_exhausted": RequestsExhaustedConstraint(
+                    "requests_exhausted": RequestsExhaustedConstraint(  # type: ignore[dict-item]
                         num_requests=count
                     )
                 },