(self, sid: str)
| 205 | self.queued_seqs.pop(sid) |
| 206 | |
| 207 | def remove(self, sid: str): |
| 208 | if self.in_decoding(sid): |
| 209 | self.remove_decoding(sid) |
| 210 | elif self.in_prefilling(sid): |
| 211 | self.remove_prefilling(sid) |
| 212 | elif self.in_queued(sid): |
| 213 | self.remove_queued(sid) |
| 214 | else: |
| 215 | raise ValueError(f"Sequence {sid} not found") |
| 216 | |
| 217 | def in_decoding(self, sid: str) -> bool: |
| 218 | return sid in self.decoding_seqs |
no test coverage detected