convert h vector to matrix with single column

Deep-MI · Dec 15, 2024 · 1cce152 · 1cce152
1 parent 9d93750
commit 1cce152
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lapy/solver.py b/lapy/solver.py
@@ -705,6 +705,8 @@ def poisson(self, h=0.0, dtup=(), ntup=()):  # poissonSolver
             raise ValueError(
                 "h should be either scalar or column vector with row num of A"
             )
+        if h.ndim == 1:
+            h = h[:,np.newaxis]
         # create vector d
         didx = []
         dvec = []