Refactor

[moe.git] / t / moe / pipeline.py

diff --git a/t/moe/pipeline.py b/t/moe/pipeline.py
index 1b2d98a581fbf9e24a94120e273ec27a5d520900..df4b229a1000f3a410cd06569e1343e43283bfdc 100644 (file)
--- 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))