wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-06-30 16:50:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

consistent size checking

Browse Source
This commit is contained in:
Lindsey Heagy
2015-05-28 08:39:49 -07:00
parent 369694335a
commit ec7ed8a585
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Problem.py
+2 -2
View File
@@ -160,7 +160,7 @@ class Fields(object):
assert hasattr(self, func), 'The alias field function is a string, but it does not exist in the Fields class.'
func = getattr(self, func)
out = func(self._fields[alias][:,ind], srcII)
if isinstance(out, np.ndarray) and (out.shape[0] == out.size or out.ndim == 1):
if out.shape[0] == out.size or out.ndim == 1:
out = Utils.mkvc(out,2)
return out
@@ -263,7 +263,7 @@ class TimeFields(Fields):
for i, TIND_i in enumerate(timeII):
fieldI = pointerFields[:,:,i]
if fieldI.shape[0] == fieldI.size:
fieldI = Utils.mkvc(fieldI,2)
fieldI = Utils.mkvc(fieldI,1)
out[i] = func(fieldI, srcII, TIND_i)
if out[i].ndim == 1:
out[i] = out[i][:,np.newaxis,np.newaxis]