type cast b in solver so that it is not an object if we add zero

2026-06-30 06:47:05 +08:00 · 2015-11-24 14:38:58 -08:00
parent 71a5c8b2c1
commit 5bd9209d6a
1 changed files with 8 additions and 0 deletions
@@ -37,12 +37,20 @@ def SolverWrapD(fun, factorize=True, checkAccuracy=True, accuracyTol=1e-6):
        if len(b.shape) == 1 or b.shape[1] == 1:
            b = b.flatten()
            # Just one RHS
+
+            if b.dtype is np.dtype('O'):
+                b = b.astype(type(b[0]))
+
            if factorize:
                X = self.solver.solve(b, **self.kwargs)
            else:
                X = fun(self.A, b, **self.kwargs)
        else: # Multiple RHSs
+            if b.dtype is np.dtype('O'):
+                b = b.astype(type(b[0,0]))
+
            X = np.empty_like(b)
+
            for i in range(b.shape[1]):
                if factorize:
                    X[:,i] = self.solver.solve(b[:,i])