redirect backwards comp

mlens/ensemble/tests/test_stacking_ensemble.py

@@ -4,8 +4,6 @@
 """
 
 from __future__ import division, print_function, with_statement
-from contextlib import redirect_stderr
-import os
 
 import numpy as np
 from pandas import DataFrame
@@ -26,6 +24,13 @@
 
 import warnings
 
+import os
+try:
+    from contextlib import redirect_stderr
+except ImportError:
+    from mlens.externals.fixes import redirect as redirect_stderr
+
+
 # Training data
 np.random.seed(100)
 X = np.random.random((1000, 10))

mlens/externals/fixes.py

@@ -0,0 +1,22 @@
+"""ML-ENSEMBLE
+
+Collection of simpler fixes occassionally necessary for backwards
+compatibility.
+"""
+
+import sys
+from contextlib import contextmanager
+
+
+@contextmanager
+def redirect(target, file=sys.stderr):
+    """Replication of ``redirect_stderr`` or ``redirect_stdout``.
+
+    :source: http://stackoverflow.com/questions/4675728/redirect-stdout-to-a-file-in-python/22434262#22434262
+    """
+    original, file = file, target
+
+    try:
+        yield target
+    finally:
+        file = original

mlens/model_selection/tests/test_model_selection.py

@@ -4,8 +4,6 @@
 """
 
 from __future__ import division, print_function
-from contextlib import redirect_stderr
-import os
 
 import numpy as np
 
@@ -22,6 +20,13 @@
 import warnings
 import subprocess
 
+import os
+try:
+    from contextlib import redirect_stderr
+except ImportError:
+    from mlens.externals.fixes import redirect as redirect_stderr
+
+
 # training data
 np.random.seed(100)
 X = np.random.random((1000, 10))

mlens/preprocessing/tests/test_feature_engineering.py

@@ -4,8 +4,6 @@
 """
 
 from __future__ import division, print_function
-from contextlib import redirect_stderr
-import os
 
 import numpy as np
 
@@ -23,6 +21,13 @@
 
 import warnings
 
+import os
+try:
+    from contextlib import redirect_stderr
+except ImportError:
+    from mlens.externals.fixes import redirect as redirect_stderr
+
+
 SEED = 100
 np.random.seed(SEED)