Fix typing

DiamondLightSource · Apr 12, 2024 · 0ab2b72 · 0ab2b72
1 parent a70bac7
commit 0ab2b72
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/odin_fastcs/odin_controller.py b/src/odin_fastcs/odin_controller.py
@@ -82,7 +82,7 @@ def __init__(
         self._api_prefix = api_prefix
         self._process_prefix = process_prefix
         self._path = process_prefix
-        self._cached_config_params = {}
+        self._cached_config_params: dict[str, Any] = {}
         # used to determine if we need to connect the param tree or C++ params
         self._has_param_tree = param_tree
         self._has_process_params = process_params

diff --git a/src/odin_fastcs/util.py b/src/odin_fastcs/util.py
@@ -33,7 +33,7 @@ def flatten_dict(
 def unflatten_dict(
     dd: dict[str, Any], separator: str = "/", reverse_indexing: bool = False
 ) -> Mapping[str, Any]:
-    output: Mapping[str, Any] = {}
+    output: dict[str, Any] = {}
     for key, val in dd.items():
         key_parts = key.split(separator)
         if reverse_indexing:
@@ -61,9 +61,9 @@ def map_short_name_to_path_and_value(
 
     def get_name_mapping(
         tree: Mapping[str, Any],
-        name_parts: list[str] = None,
+        name_parts: list[str] | None = None,
         parts_needed: int = 1,
-        mapping: dict[str, str] = None,
+        mapping: dict[str, str] | None = None,
     ) -> dict[str, str]:
         if mapping is None:
             mapping = {}