Shape refactor: Fix dict type

tum-pbs · Dec 22, 2024 · 9a2c78a · 9a2c78a
1 parent 73ab07d
commit 9a2c78a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phiml/math/_shape.py b/phiml/math/_shape.py
@@ -2341,7 +2341,7 @@ def pure_merge(*shapes: Shape, allow_varying_sizes: bool) -> Shape:
         return EMPTY_SHAPE
     if len(non_empty) == 1:
         return non_empty[0]
-    dims = dict[str, Dim]()
+    dims = {}
     for dim in all_dims:
         if dim.name not in dims:
             dims[dim.name] = dim