From cfcf741a0be238bd142dac6052af6de348b176a8 Mon Sep 17 00:00:00 2001 From: rowanc1 Date: Thu, 15 May 2014 12:21:24 -0700 Subject: [PATCH] Clean up p indexing --- simpegEM/TDEM/TDEM_b.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/simpegEM/TDEM/TDEM_b.py b/simpegEM/TDEM/TDEM_b.py index 18645575..fbac0813 100644 --- a/simpegEM/TDEM/TDEM_b.py +++ b/simpegEM/TDEM/TDEM_b.py @@ -162,7 +162,9 @@ class ProblemTDEM_b(BaseTDEMProblem): y_b = sol if self.survey.nTx == 1: y_b = mkvc(y_b) - y_e = self.MeSigmaI*(self.mesh.edgeCurl.T*(self.MfMui*y_b)) - self.MeSigmaI*p[:,'e',tInd+1] + y_e = self.MeSigmaI*(self.mesh.edgeCurl.T*(self.MfMui*y_b)) + if 'e' in p: + y_e = y_e - self.MeSigmaI*p[:,'e',tInd+1] return {'b':y_b, 'e':y_e} return self.forward(m, AhRHS, AhCalcFields)