linting

masci · May 2, 2024 · 1cbd2b5 · 1cbd2b5
1 parent f411ae7
commit 1cbd2b5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/banks/extensions/generate.py b/src/banks/extensions/generate.py
@@ -14,7 +14,7 @@
 SYSTEM_PROMPT = Prompt("{{canary_word}} You are a helpful assistant.")
 
 
-def generate(model_name: str):
+def generate(model_name: str):  # noqa
     """
     `generate` can be used to call the LiteLLM API passing the tag text as a prompt and get back some content.
 

diff --git a/src/banks/prompt.py b/src/banks/prompt.py
@@ -47,7 +47,7 @@ def from_template(cls, name: str) -> "BasePrompt":
         ```
         .
         └── templates
-            └── foo.jinja
+            └── foo.jinja
         ```
 
         The code would be the following: