diff --git a/simpegMT/ProblemMT.py b/simpegMT/ProblemMT.py index 0d1a923e..e4617158 100644 --- a/simpegMT/ProblemMT.py +++ b/simpegMT/ProblemMT.py @@ -108,8 +108,9 @@ class MTProblem(Problem.BaseProblem): # RHS, CalcFields = self.getRHS(freq,m_back), self.calcFields F = FieldsMT(self.mesh, self.survey) - startTime = time.time() + def solveAtFreq(self,F,freq): + startTime = time.time() print 'Starting work for {:.3e}'.format(freq) sys.stdout.flush() A = self.getA(freq) diff --git a/simpegMT/SurveyMT.py b/simpegMT/SurveyMT.py index 940cfaab..f6e8d042 100644 --- a/simpegMT/SurveyMT.py +++ b/simpegMT/SurveyMT.py @@ -1,5 +1,6 @@ from SimPEG import Survey, Utils, Problem, np, sp, mkvc from scipy.constants import mu_0 +import sys class RxMT(Survey.BaseRx):