fix in ortho_left and ortho_right

PGelss · Jul 17, 2024 · 4c38363 · 4c38363
1 parent 7a8bc2d
commit 4c38363
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scikit_tt/tensor_train.py b/scikit_tt/tensor_train.py
@@ -1151,6 +1151,8 @@ def ortho_left(self, start_index: int=0,
                                     max_rank_tf = False
                         if max_rank_tf:
                             max_ranks = max_rank
+                    else:
+                        max_rank_tf = False
 
                 if max_rank_tf:
 
@@ -1255,6 +1257,8 @@ def ortho_right(self, start_index: Optional[int]=None,
                                     max_rank_tf = False
                         if max_rank_tf:
                             max_ranks = max_rank
+                    else:
+                        max_rank_tf = False
 
                 if max_rank_tf: