X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=t%2Fmoe%2Fpipeline.py;h=df4b229a1000f3a410cd06569e1343e43283bfdc;hb=420a49640eb8a89b578c390a939bc0ba818f9a7e;hp=1b2d98a581fbf9e24a94120e273ec27a5d520900;hpb=184a1d862089a66f7cab486c607f428b098c94af;p=moe.git diff --git a/t/moe/pipeline.py b/t/moe/pipeline.py index 1b2d98a..df4b229 100644 --- a/t/moe/pipeline.py +++ b/t/moe/pipeline.py @@ -1,22 +1,24 @@ #!/usr/bin/env python import bisect + import moe +from moe.logs import * class MoePipeError(moe.MoeError): """Failure of the MoePipeline.""" class MoeAbortPipeline(Exception): - def __init__(self, skip_to=999): + def __init__(self, skip_to=None): self.skip_to = skip_to + class Pipeline: """Moe pipeline.""" - def __init__(self, e, name, skip_to = 70): + def __init__(self, name, skip_to = 700): # e is Eval - self.e = e self.pipe = [] self.index = -1 self.name = name @@ -58,14 +60,14 @@ class Pipeline: while self.index < len(self.pipe): (pri,name,fun) = self.pipe[self.index] if pri >= min_pri: - self.e.log.debug("Pipeline %r:%d running: %s" % (self.name, pri, name)) + log.debug("Pipeline %r:%d running: %s" % (self.name, pri, name)) try: fun(*args, **kwargs) except MoeAbortPipeline, err: min_pri = self.skip_to else: - self.e.log.debug("Pipeline %r:d skipping: %s" % (self.name, pri, name)) + log.debug("Pipeline %r:d skipping: %s" % (self.name, pri, name)) self.index += 1 self.index = -1 - self.e.log.debug("Pipeline %r finished" % (self.name)) + log.debug("Pipeline %r finished" % (self.name))