mypy fixes

Author: stevebachmeier

src/layered_config_tree/utilities.py

@@ -8,6 +8,7 @@
 
 """
 
+from collections.abc import Hashable
 from pathlib import Path
 from typing import Any
 
@@ -59,7 +60,9 @@ def load_yaml(data: str | Path) -> dict[str, Any]:
 class SafeLoader(yaml.SafeLoader):
     """A yaml.SafeLoader that restricts duplicate keys."""
 
-    def construct_mapping(self, node, deep=False):
+    def construct_mapping(
+        self, node: yaml.MappingNode, deep: bool = False
+    ) -> dict[Hashable, Any]:
         """Constructs the standard mapping after checking for duplicates.
 
         Raises
@@ -79,7 +82,7 @@ def construct_mapping(self, node, deep=False):
         """
         mapping = []
         for key_node, _value_node in node.value:
-            key = self.construct_object(key_node, deep=deep)
+            key = self.construct_object(key_node, deep=deep)  # type: ignore[no-untyped-call]
             if key in mapping:
                 raise DuplicatedConfigurationError(
                     f"Duplicate key detected at same level of YAML: {key}. Resolve duplicates and try again.",

tests/test_ingestion.py

@@ -63,11 +63,11 @@ def test_load_yaml_file(tmp_path: Path, path_type: type[str | Path]) -> None:
 def test_load_yaml_duplicates_raise(
     duplicates: bool, load_from_file: bool, tmp_path: Path
 ) -> None:
-    test_yaml = TEST_YAML_DUPLICATE_KEYS if duplicates else TEST_YAML_ONE
+    test_str: str = TEST_YAML_DUPLICATE_KEYS if duplicates else TEST_YAML_ONE
     if load_from_file:
         tmp_file = tmp_path / "test_dupliate_keys.yaml"
-        tmp_file.write_text(test_yaml)
-        test_yaml = tmp_file
+        tmp_file.write_text(test_str)
+    test_yaml = tmp_file if load_from_file else test_str
 
     lct = LayeredConfigTree()