more linting

pyproject.toml

@@ -170,5 +170,8 @@ exclude_lines = [
 ]
 
 [[tool.mypy.overrides]]
-module = "simplemma.*"
+module = [
+  "simplemma.*",
+  "litellm.*"
+]
 ignore_missing_imports = true

src/banks/extensions/generate.py

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2023-present Massimiliano Pippi <[email protected]>
 #
 # SPDX-License-Identifier: MIT
-from typing import ClassVar, cast
+from typing import cast
 
 from jinja2 import nodes
 from jinja2.ext import Extension
@@ -22,7 +22,7 @@ class GenerateExtension(Extension):
     """
 
     # a set of names that trigger the extension.
-    tags: ClassVar[set] = {"generate"}
+    tags = {"generate"}  # noqa
 
     def parse(self, parser):
         # We get the line number of the first token so that we can give

src/banks/extensions/inference_endpoint.py

@@ -3,7 +3,6 @@
 # SPDX-License-Identifier: MIT
 import html
 import os
-from typing import ClassVar
 
 import requests
 from jinja2 import nodes
@@ -23,7 +22,7 @@ class HFInferenceEndpointsExtension(Extension):
     """
 
     # a set of names that trigger the extension.
-    tags: ClassVar[set] = {"inference_endpoint"}
+    tags = {"inference_endpoint"}  # noqa
 
     def parse(self, parser):
         # We get the line number of the first token so that we can give