diff --git a/pylot/core/active/seismicshot.py b/pylot/core/active/seismicshot.py index 3195af4a..63f1911c 100644 --- a/pylot/core/active/seismicshot.py +++ b/pylot/core/active/seismicshot.py @@ -327,6 +327,7 @@ class SeismicShot(object): def pickParallel(self, folm, method = 'hos', aicwindow = (10, 0)): import multiprocessing + from pylot.core.util.utils import worker self.setFolm(folm) self.setMethod(method) @@ -337,11 +338,10 @@ class SeismicShot(object): traceIDs = self.getTraceIDlist() - picks = pool.map(self.pickTrace, traceIDs) + picks = worker(self.pickTrace, traceIDs, maxthreads) for traceID, pick in picks: self.setPick(traceID, pick) - def pickTrace(self, traceID): '''