diff --git a/simpegEM/TDEM/FieldsTDEM.py b/simpegEM/TDEM/FieldsTDEM.py index 0a64881a..c6d59d58 100644 --- a/simpegEM/TDEM/FieldsTDEM.py +++ b/simpegEM/TDEM/FieldsTDEM.py @@ -29,9 +29,11 @@ class FieldsTDEM(object): self.set_e(newFields['e'], tInd) def fieldVec(self): - u = np.ndarray((0,self.nTx)) + u = np.ndarray((0, self.nTx)) for i in range(self.nTimes): u = np.r_[u, self.get_b(i), self.get_e(i)] + if self.nTx == 1: + u = u.flatten() return u #################################################### @@ -64,4 +66,4 @@ class FieldsTDEM(object): if self.e is None: self.e = np.zeros((self.nTimes, np.sum(self.mesh.nE), self.nTx)) self.e[:] = np.nan - self.e[ind,:,:] = e \ No newline at end of file + self.e[ind,:,:] = e