Merge pull request #92 from simpeg/master

merge fix in fields size to removeSrcType
simpeg · May 7, 2015 · f832418 · f832418
2 parents 6f487e1 + 4bda4b6
commit f832418
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/SimPEG/Problem.py b/SimPEG/Problem.py
@@ -160,8 +160,7 @@ def _getField(self, name, ind):
                 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 out.shape[1] == 1:
+        if out.shape[0] == out.size:
             out = Utils.mkvc(out)
         return out