Tweak type annotation.

dbt-labs · Jan 23, 2024 · 3a96528 · 3a96528
1 parent b9c0c2d
commit 3a96528
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dbt_common/clients/_jinja_blocks.py b/dbt_common/clients/_jinja_blocks.py
@@ -1,6 +1,6 @@
 import re
 from collections import namedtuple
-from typing import Iterator, Optional, List
+from typing import Iterator, Optional, List, Set
 
 from dbt_common.exceptions import (
     BlockDefinitionNotAtTopError,
@@ -294,7 +294,7 @@ def is_current_end(self, tag: Tag) -> bool:
         )
 
     def find_blocks(
-        self, allowed_blocks: Optional[set[str]] = None, collect_raw_data: bool = True
+        self, allowed_blocks: Optional[Set[str]] = None, collect_raw_data: bool = True
     ) -> Iterator[BlockData | BlockTag]:
         """Find all top-level blocks in the data."""
         if allowed_blocks is None:
@@ -347,6 +347,6 @@ def find_blocks(
                 yield BlockData(raw_data)
 
     def lex_for_blocks(
-        self, allowed_blocks: Optional[set[str]] = None, collect_raw_data: bool = True
+        self, allowed_blocks: Optional[Set[str]] = None, collect_raw_data: bool = True
     ) -> List[BlockData | BlockTag]:
         return list(self.find_blocks(allowed_blocks=allowed_blocks, collect_raw_data=collect_raw_data))