From 8d97f322c984d9de7830d7917ab9258f9f80f85d Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 28 May 2015 11:15:58 -0700 Subject: [PATCH] analytics return arrays --- simpegEM/Analytics/FDEM.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/simpegEM/Analytics/FDEM.py b/simpegEM/Analytics/FDEM.py index a0ec6776..3d02b589 100644 --- a/simpegEM/Analytics/FDEM.py +++ b/simpegEM/Analytics/FDEM.py @@ -99,5 +99,15 @@ def AnalyticMagDipoleWholeSpace(XYZ, srcLoc, sig, f, m=1., orientation='X'): Bx = mu_0*Hx By = mu_0*Hy Bz = mu_0*Hz + + if Bx.ndim is 1: + Bx = Utils.mkvc(Bx,2) + + if By.ndim is 1: + By = Utils.mkvc(By,2) + + if Bz.ndim is 1: + Bz = Utils.mkvc(Bz,2) + return Bx, By, Bz