From 25acc67c43f7b440683b9cabe4516494279f2d4a Mon Sep 17 00:00:00 2001 From: GudniRos Date: Mon, 6 Apr 2015 10:42:46 -0700 Subject: [PATCH] Added time string printing --- simpegMT/ProblemMT.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/simpegMT/ProblemMT.py b/simpegMT/ProblemMT.py index 5540105d..429085fe 100644 --- a/simpegMT/ProblemMT.py +++ b/simpegMT/ProblemMT.py @@ -1,7 +1,7 @@ from SimPEG import Survey, Problem, Utils, Models, np, sp, SolverLU as SimpegSolver from scipy.constants import mu_0 from SurveyMT import SurveyMT, FieldsMT -import multiprocessing, sys +import multiprocessing, sys, time def omega(freq): """Change frequency to angular frequency, omega""" @@ -108,6 +108,7 @@ 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): print 'Starting work for {:.3e}'.format(freq) sys.stdout.flush() @@ -135,7 +136,7 @@ class MTProblem(Problem.BaseProblem): pool.map(solveAtFreq,self.survey.freqs) pool.close() pool.join() - + print 'Ran for {:f} seconds'.format(time.time()-startTime) return F