Fixed pipeline.insert args order

[eval.git] / t / moe / util.py

diff --git a/t/moe/util.py b/t/moe/util.py
index e083f0fb6f84dbfe285d342c8482036211d05d2e..d287ff864c1230709b3172d658af4e1f3f157297 100644 (file)
--- a/t/moe/util.py
+++ b/t/moe/util.py
@@ -1,21 +1,22 @@
 #!/usr/bin/env python
 
 import os
 #!/usr/bin/env python
 
 import os

+import sys

 import os.path
 import shutil
 
 def mkdir_tree(name):
     try:
        os.mkdir(name)  
 import os.path
 import shutil
 
 def mkdir_tree(name):
     try:
        os.mkdir(name)  

-    except OSError, e:
-       if e.errno == os.errno.ENOENT:
+    except OSError, err:
+       if err.errno == os.errno.ENOENT:

            head, tail = os.path.split(name)
            mkdir_tree(head)
            os.mkdir(name)
            head, tail = os.path.split(name)
            mkdir_tree(head)
            os.mkdir(name)

-       elif e.errno == os.errno.EEXIST:
+       elif err.errno == os.errno.EEXIST:

            pass
        else:
            pass
        else:

-           raise e
+           raise err

 
 def link_or_copy(src, dest):
     try:
 
 def link_or_copy(src, dest):
     try:


@@ -33,3 +34,10 @@ def remove_tree_contents(dir):
                shutil.rmtree(os.path.join(dir, f))
            else:
                raise err
                shutil.rmtree(os.path.join(dir, f))
            else:
                raise err

+
+def load_module(modname):
+    """Return the module `modname` (full name) if loaded, or try to import it. 
+    Returns the module or raises `ImportError`."""
+    if not sys.modules.has_key(modname):
+       __import__(modname)
+    return sys.modules[modname]