From 4a9198ade77157633791e621897478e1b3572d9b Mon Sep 17 00:00:00 2001
From: Jan Petykiewicz <jan@mpxd.net>
Date: Mon, 14 Nov 2022 12:39:29 -0800
Subject: [PATCH] allow setting initial guess

---
 meanas/fdfd/bloch.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/meanas/fdfd/bloch.py b/meanas/fdfd/bloch.py
index ea2740f..08a9654 100644
--- a/meanas/fdfd/bloch.py
+++ b/meanas/fdfd/bloch.py
@@ -480,6 +480,7 @@ def eigsolve(
         tolerance: float = 1e-20,
         max_iters: int = 10000,
         reset_iters: int = 100,
+        y0: Optional[ArrayLike] = None,
         callback: Optional[Callable[[...], None]] = None,
         ) -> Tuple[NDArray[numpy.complex128], NDArray[numpy.complex128]]:
     """
@@ -498,6 +499,7 @@ def eigsolve(
         max_iters: TODO
         reset_iters: TODO
         callback: TODO
+        y0: TODO, initial guess
 
     Returns:
         `(eigenvalues, eigenvectors)` where `eigenvalues[i]` corresponds to the
@@ -526,7 +528,6 @@ def eigsolve(
     #prev_theta = 0.5
     D = numpy.zeros(shape=y_shape, dtype=complex)
 
-    y0 = None
     if y0 is None:
         Z = numpy.random.rand(*y_shape) + 1j * numpy.random.rand(*y_shape)
     else: