format

neulab · Jul 9, 2023 · 272fc73 · 272fc73
1 parent ccded32
commit 272fc73
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/llm_client/__init__.py b/llm_client/__init__.py
@@ -55,7 +55,9 @@ def prompt(
         Returns:
         """
         if isinstance(text, str):
-            return self.prompt([text], max_new_tokens, output_scores, output_hidden_states, **kwargs)
+            return self.prompt(
+                [text], max_new_tokens, output_scores, output_hidden_states, **kwargs
+            )
 
         # TODO: Check for max length limit to avoid OOMs