[FIX] Minor bug fixes (#1035)

* [FIX] Minor bug fixes * Address review comments
2023-09-13 16:38:12 -07:00
parent ab019eea75
commit f04908cae7
2 changed files with 4 additions and 3 deletions
--- a/vllm/sequence.py
+++ b/vllm/sequence.py
@@ -350,7 +350,7 @@ class SequenceOutputs:

    def __eq__(self, other: object) -> bool:
        if not isinstance(other, SequenceOutputs):
-            return NotImplementedError()
+            raise NotImplementedError()
        return (self.parent_seq_id == other.parent_seq_id
                and self.output_token == other.output_token
                and self.logprobs == other.logprobs)