whitespace colon suppress

5 months ago · b162e76ca0
parent ea649b943d
commit b162e76ca0
1 changed files with 1 additions and 1 deletions
--- a/swarms/tools/logits_processor.py
+++ b/swarms/tools/logits_processor.py
@ -48,7 +48,7 @@ class NumberStoppingCriteria(StoppingCriteria):
        scores: torch.FloatTensor,
    ) -> bool:
        decoded = self.tokenizer.decode(
-            input_ids[0][self.prompt_length :],
+            input_ids[0][self.prompt_length :],   # noqa: E203
            skip_special_tokens=True,
        )  # noqa: E203