typing fix

quicophy · Feb 14, 2025 · d92a244 · d92a244
1 parent 4cf61a7
commit d92a244
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mdopt/mps/canonical.py b/mdopt/mps/canonical.py
@@ -127,7 +127,7 @@ def phys_dimensions(self) -> List[int]:
         return [self.tensors[i].shape[1] for i in range(self.num_sites)]
 
     @property
-    def all_dimensions(self) -> List[Tuple[int, int, int]]:
+    def all_dimensions(self) -> List[Tuple[int, ...]]:
         """
         Returns the list of all dimensions of the MPS.
         """

diff --git a/mdopt/mps/explicit.py b/mdopt/mps/explicit.py
@@ -118,7 +118,7 @@ def phys_dimensions(self) -> List[int]:
         return [self.tensors[i].shape[1] for i in range(self.num_sites)]
 
     @property
-    def all_dimensions(self) -> List[Tuple[int, int, int]]:
+    def all_dimensions(self) -> List[Tuple[int, ...]]:
         """
         Returns the list of all dimensions of the MPS.
         """