[Misc] Add missing return type annotations in outputs.py (#52145)
Signed-off-by: Vineeta Tiwari <vineeta.tiwari2@ibm.com> Co-authored-by: Vineeta Tiwari <vineeta.tiwari2@ibm.com>
Vineeta Tiwari committed
Aug 13, 2026 at 18:28 UTC
015660da91a6710465d2a24eec3394c178fb7181
1 file changed
+13
-7
vllm/outputs.py
+13
-7
@@ -251,7 +251,7 @@ class PoolingRequestOutput(Generic[_O]):
251
self.finished = finished
252
self.outputs = outputs
253
254
- def __repr__(self):
254
+ def __repr__(self) -> str:
255
return (
256
f"{type(self).__name__}(request_id={self.request_id!r}, "
257
f"outputs={self.outputs!r}, "
@@ -273,7 +273,7 @@ class EmbeddingOutput:
273
embedding: list[float]
274
275
@staticmethod
276
- def from_base(pooling_output: PoolingOutput):
276
+ def from_base(pooling_output: PoolingOutput) -> "EmbeddingOutput":
277
pooled_data = pooling_output.data
278
if pooled_data.ndim != 1:
279
raise ValueError("pooled_data should be a 1-D embedding vector")
@@ -290,7 +290,9 @@ class EmbeddingOutput:
290
291
class EmbeddingRequestOutput(PoolingRequestOutput[EmbeddingOutput]):
292
@staticmethod
293
- def from_base(request_output: PoolingRequestOutput):
293
+ def from_base(
294
+ request_output: PoolingRequestOutput,
295
+ ) -> "EmbeddingRequestOutput":
296
return EmbeddingRequestOutput(
297
request_id=request_output.request_id,
298
outputs=EmbeddingOutput.from_base(request_output.outputs),
@@ -312,7 +314,7 @@ class ClassificationOutput:
314
probs: list[float]
315
316
@staticmethod
315
- def from_base(pooling_output: PoolingOutput):
317
+ def from_base(pooling_output: PoolingOutput) -> "ClassificationOutput":
318
# pooling_output shape: (num_classes)
319
pooled_data = pooling_output.data
320
if pooled_data.ndim != 1:
@@ -330,7 +332,9 @@ class ClassificationOutput:
332
333
class ClassificationRequestOutput(PoolingRequestOutput[ClassificationOutput]):
334
@staticmethod
333
- def from_base(request_output: PoolingRequestOutput):
335
+ def from_base(
336
+ request_output: PoolingRequestOutput,
337
+ ) -> "ClassificationRequestOutput":
338
return ClassificationRequestOutput(
339
request_id=request_output.request_id,
340
outputs=ClassificationOutput.from_base(request_output.outputs),
@@ -351,7 +355,7 @@ class ScoringOutput:
355
score: float
356
357
@staticmethod
354
- def from_base(pooling_output: PoolingOutput):
358
+ def from_base(pooling_output: PoolingOutput) -> "ScoringOutput":
359
# pooling_output shape:
360
# classify task: (num_classes) num_classes == 1
361
# embed task: a scalar value
@@ -367,7 +371,9 @@ class ScoringOutput:
371
372
class ScoringRequestOutput(PoolingRequestOutput[ScoringOutput]):
373
@staticmethod
370
- def from_base(request_output: PoolingRequestOutput):
374
+ def from_base(
375
+ request_output: PoolingRequestOutput,
376
+ ) -> "ScoringRequestOutput":
377
return ScoringRequestOutput(
378
request_id=request_output.request_id,
379
outputs=ScoringOutput.from_base(request_output.outputs),