def clear()

in text-generation-inference/server/text_generation_server/generator.py [0:0]

• 5 lines of code
• 4 McCabe index (conditional complexity)

    def clear(self, batch_id: Optional[int] = None):
        """Remove a subset or all requests from the generator"""
        keep_ids = []
        if batch_id is not None:
            keep_ids = [slot.id for slot in self.slots if slot.batch_id != batch_id]
        return self._clear(keep_ids)