From d92a2447d2629675371b265ac5470fb4deb4c0f0 Mon Sep 17 00:00:00 2001 From: meandmytram Date: Thu, 13 Feb 2025 20:41:10 -0500 Subject: [PATCH] typing fix --- mdopt/mps/canonical.py | 2 +- mdopt/mps/explicit.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mdopt/mps/canonical.py b/mdopt/mps/canonical.py index 9d0c1b2..07ebefe 100644 --- 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 index 596d132..4360c1f 100644 --- 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. """