fix: types

src/inline_snapshot/_change.py

@@ -12,7 +12,6 @@
 
 from asttokens.util import Token
 from executing.executing import EnhancedAST
-from executing.executing import Source
 from inline_snapshot._source_file import SourceFile
 
 from ._rewrite_code import ChangeRecorder
@@ -101,7 +100,7 @@ def brace_tokens(source, node) -> TokenRange:
 
 
 def generic_sequence_update(
-    source: Source,
+    source: SourceFile,
     parent: Union[ast.List, ast.Tuple, ast.Dict, ast.Call],
     brace_tokens: TokenRange,
     parent_elements: List[Union[TokenRange, None]],
@@ -166,7 +165,7 @@ def apply_all(all_changes: List[Change]):
     by_parent: Dict[
         EnhancedAST, List[Union[Delete, DictInsert, ListInsert, CallArg]]
     ] = defaultdict(list)
-    sources: Dict[EnhancedAST, Source] = {}
+    sources: Dict[EnhancedAST, SourceFile] = {}
 
     for change in all_changes:
         if isinstance(change, Delete):

src/inline_snapshot/_rewrite_code.py

@@ -124,7 +124,7 @@ def _replace(self, filename, range, new_contend):
 
 
 class SourceFile:
-    def __init__(self, filename):
+    def __init__(self, filename: pathlib.Path):
         self.replacements: list[Replacement] = []
         self.filename = filename
         self.source = self.filename.read_text("utf-8")

src/inline_snapshot/_source_file.py

@@ -1,6 +1,7 @@
 import tokenize
 from pathlib import Path
 
+from executing import Source
 from inline_snapshot._format import format_code
 from inline_snapshot._unmanaged import is_dirty_equal
 from inline_snapshot._utils import normalize
@@ -11,6 +12,7 @@
 
 
 class SourceFile:
+    _source = Source
 
     def __init__(self, source):
         if isinstance(source, SourceFile):

src/inline_snapshot/_unmanaged.py

@@ -17,7 +17,7 @@ def is_dirty_equal(value):
 
 
 def update_allowed(value):
-    return not (is_dirty_equal(value) or isinstance(value, (Is, Snapshot)))
+    return not (is_dirty_equal(value) or isinstance(value, (Is, Snapshot)))  # type: ignore
 
 
 def is_unmanaged(value):

tests/conftest.py

@@ -20,6 +20,7 @@
 from inline_snapshot._format import format_code
 from inline_snapshot._inline_snapshot import Flags
 from inline_snapshot._rewrite_code import ChangeRecorder
+from inline_snapshot._types import Category
 from inline_snapshot.testing._example import snapshot_env
 
 pytest_plugins = "pytester"
@@ -53,7 +54,7 @@ def w(source_code, *, flags="", reported_flags=None, number=1):
 
 
 @pytest.fixture()
-def source(tmp_path):
+def source(tmp_path: Path):
     filecount = 1
 
     @dataclass
@@ -64,8 +65,8 @@ class Source:
         number_snapshots: int = 0
         number_changes: int = 0
 
-        def run(self, *flags):
-            flags = Flags({*flags})
+        def run(self, *flags_arg: Category):
+            flags = Flags({*flags_arg})
 
             nonlocal filecount
             filename: Path = tmp_path / f"test_{filecount}.py"