Fix Shape member type hints

tum-pbs · Dec 14, 2024 · 88f5813 · 88f5813
1 parent aca8e75
commit 88f5813
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/phiml/math/_shape.py b/phiml/math/_shape.py
@@ -66,15 +66,15 @@ def __init__(self, sizes: tuple, names: tuple, types: tuple, item_names: tuple):
         See Also:
             `Shape.get_size()`, `Shape.size`, `Shape.shape`.
         """
-        self.names: Tuple[str] = names
+        self.names: Tuple[str, ...] = names
         """
         Ordered dimension names as `tuple[str]`.
         
         See Also:
             `Shape.name`.
         """
-        self.types: Tuple[str] = types  # undocumented, may be private
-        self.item_names: Tuple[Optional[Tuple[str, ...]]] = (None,) * len(sizes) if item_names is None else item_names  # undocumented
+        self.types: Tuple[str, ...] = types  # undocumented, may be private
+        self.item_names: Tuple[Optional[Tuple[str, ...]], ...] = (None,) * len(sizes) if item_names is None else item_names  # undocumented
         if DEBUG_CHECKS:
             assert len(sizes) == len(names) == len(types) == len(item_names), f"sizes={sizes}, names={names}, types={types}, item_names={item_names}"
             assert len(set(names)) == len(names), f"Duplicate dimension names: {names}"