Fix MyPy

.github/workflows/code-style.yml

@@ -11,10 +11,6 @@ jobs:
     steps:
       - name: 'Checkout sources'
         uses: actions/checkout@v2
-      - name: 'Format code with black'
-        run: |
-          pip install black
-          black .
       - name: 'Lint code with Ruff'
         run: |
           pip install ruff
@@ -23,6 +19,10 @@ jobs:
         run: |
           pip install mypy
           mypy --ignore-missing-imports codelimit/
+      - name: 'Format code with black'
+        run: |
+          pip install black
+          black .
       - uses: stefanzweifel/git-auto-commit-action@v4
         with:
           commit_message: 'Code style fixes'

codelimit/common/scope/scope_utils.py

@@ -129,7 +129,7 @@ def has_curly_suffix(tokens: list[Token], index):
     return index < len(tokens) - 1 and tokens[index + 1].is_symbol("{")
 
 
-def get_headers(tokens: list[Token], pattern: list[TokenPredicate]):
+def get_headers(tokens: list[Token], pattern: list[TokenPredicate | str]):
     matches = Matcher(pattern).match(tokens)
     result = []
     for m in matches:

codelimit/common/token_matching/Matcher.py

@@ -8,7 +8,7 @@
 
 
 class Matcher:
-    def __init__(self, pattern: Union[TokenPredicate | str, list[TokenPredicate | str]]):
+    def __init__(self, pattern: Union[TokenPredicate, str, list[TokenPredicate | str]]):
         if not isinstance(pattern, list):
             if isinstance(pattern, TokenPredicate):
                 self._pattern = [pattern]