wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-23 13:10:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

update solver utils.

Browse Source
This commit is contained in:
rowanc1
2014-05-16 18:46:31 -07:00
parent f3170a8a7f
commit 26340fe90f
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Utils/SolverUtils.py
+6 -3
View File
@@ -42,10 +42,9 @@ def DSolverWrap(fun, factorize=True, checkAccuracy=True, accuracyTol=1e-6):
return X
def clean(self):
if hasattr(self.solver, 'clean'):
if factorize and hasattr(self.solver, 'clean'):
return self.solver.clean()
def __mul__(self, val):
if type(val) is np.ndarray:
return self.solve(val)
@@ -87,9 +86,13 @@ def ISolverWrap(fun, checkAccuracy=True, accuracyTol=1e-5):
_checkAccuracy(self.A, b, X, accuracyTol)
return X
def clean(self):
if hasattr(self.solver, 'clean'):
return self.solver.clean()
def __mul__(self, val):
if type(val) is np.ndarray:
return self.solve(val)
raise TypeError('Can only multiply by a numpy array.')
return type(fun.__name__, (object,), {"__init__": __init__, "solve": solve, "__mul__": __mul__})
return type(fun.__name__, (object,), {"__init__": __init__, "solve": solve, "clean": clean, "__mul__": __mul__})