suppress pre colon error flake8

pull/735/head
evelynmitchell 5 months ago
parent 8e0e9182ae
commit 18959f95ec

@ -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]

Loading…
Cancel
Save