From c01f1a4a5a1f4af2bb94b947fac4ec3d4b176a77 Mon Sep 17 00:00:00 2001
From: rowanc1 <rowan@visiblegeology.com>
Date: Wed, 12 Feb 2014 15:24:55 -0800
Subject: [PATCH] make fields return vector if only one Tx

---
 simpegEM/TDEM/FieldsTDEM.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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