From 18959f95ecaba017aca0f3c9c22fb1579ab9bc92 Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Wed, 27 Nov 2024 18:40:49 -0700 Subject: [PATCH] suppress pre colon error flake8 --- swarms/tools/json_former.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/swarms/tools/json_former.py b/swarms/tools/json_former.py index 01d608a5..bee6526f 100644 --- a/swarms/tools/json_former.py +++ b/swarms/tools/json_former.py @@ -117,7 +117,7 @@ class Jsonformer: response[0], skip_special_tokens=True ) - response = response[len(prompt) :] + response = response[len(prompt) :] # noqa: E203 response = response.strip().rstrip(".") self.debug("[generate_number]", response) try: @@ -136,7 +136,7 @@ class Jsonformer: prompt = self.get_prompt() self.debug("[generate_number]", prompt, is_prompt=True) response = self.llm(prompt) - response = response[len(prompt) :] + response = response[len(prompt) :] # noqa: E203 response = response.strip().rstrip(".") self.debug("[generate_number]", response) try: @@ -231,7 +231,7 @@ class Jsonformer: == input_tokens ).all() ): - response = response[0][len(input_tokens[0]) :] + response = response[0][len(input_tokens[0]) :] # noqa: E203 if response.shape[0] == 1: response = response[0] @@ -261,7 +261,7 @@ class Jsonformer: == input_tokens ).all() ): - response = response[0][len(input_tokens[0]) :] + response = response[0][len(input_tokens[0]) :] # noqa: E203 if response.shape[0] == 1: response = response[0]