Ensure that iteration is the same length for preconditioner

JanLuca · JanLuca · commit 3c44f9353111 · 2025-11-24T11:51:47.000+01:00
diff --git a/varipeps/optimization/optimizer.py b/varipeps/optimization/optimizer.py
@@ -192,7 +192,7 @@ def apply_precond(x):
         x = [
             apply_contraction_jitted(contraction, (te.tensor,), (te,), (xe,))
             + norm_grad_square * xe
-            for te, xe in zip(t_objs, x, strict=True)
+            for te, xe in zip(t_objs, x[: len(t_objs)], strict=True)
         ]
 
         return _make_1d(x)
@@ -863,7 +863,9 @@ def random_noise(a):
                             )[0]
                             for te, xe in zip(
                                 working_unitcell.get_unique_tensors(),
-                                descent_dir,
+                                descent_dir[
+                                    : working_unitcell.get_len_unique_tensors()
+                                ],
                                 strict=True,
                             )
                         ]

Original file line number	Diff line number	Diff line change
`@@ -192,7 +192,7 @@ def apply_precond(x):`
`192`	`192`	`x = [`
`193`	`193`	`apply_contraction_jitted(contraction, (te.tensor,), (te,), (xe,))`
`194`	`194`	`+ norm_grad_square * xe`
`195`		`- for te, xe in zip(t_objs, x, strict=True)`
	`195`	`+ for te, xe in zip(t_objs, x[: len(t_objs)], strict=True)`
`196`	`196`	`]`
`197`	`197`
`198`	`198`	`return _make_1d(x)`
`@@ -863,7 +863,9 @@ def random_noise(a):`
`863`	`863`	`)[0]`
`864`	`864`	`for te, xe in zip(`
`865`	`865`	`working_unitcell.get_unique_tensors(),`
`866`		`- descent_dir,`
	`866`	`+ descent_dir[`
	`867`	`+ : working_unitcell.get_len_unique_tensors()`
	`868`	`+ ],`
`867`	`869`	`strict=True,`
`868`	`870`	`)`
`869`	`871`	`]`