fix linting

src/banks/env.py

@@ -13,9 +13,9 @@ def _add_extensions(_env):
 
     For example, we use banks to manage the system prompt in `GenerateExtension`
     """
+    from .extensions.chat import ChatMessage  # pylint: disable=import-outside-toplevel
     from .extensions.generate import GenerateExtension  # pylint: disable=import-outside-toplevel
     from .extensions.inference_endpoint import HFInferenceEndpointsExtension  # pylint: disable=import-outside-toplevel
-    from .extensions.chat import ChatMessage
 
     _env.add_extension(GenerateExtension)
     _env.add_extension(HFInferenceEndpointsExtension)

src/banks/extensions/chat.py

@@ -40,7 +40,7 @@ def parse(self, parser):
         # Anything else is a parse error
         error_msg = f"Invalid syntax for chat attribute, got '{gathered}', expected role=\"value\""
         try:
-            attr_name, attr_assign, attr_value = gathered
+            attr_name, attr_assign, attr_value = gathered  # pylint: disable=unbalanced-tuple-unpacking
         except ValueError:
             raise TemplateSyntaxError(error_msg, lineno) from None
 

tests/test_prompt.py

@@ -5,7 +5,7 @@
 import regex as re
 from jinja2 import Environment
 
-from banks import AsyncPrompt, Prompt, ChatMessage
+from banks import AsyncPrompt, ChatMessage, Prompt
 from banks.cache import DefaultCache
 from banks.errors import AsyncError