fixing the last commit

Author: djarecka

nipype/interfaces/tests/test_runtime_profiler.py

@@ -14,6 +14,7 @@
 from nipype.interfaces.base import (traits, CommandLine, CommandLineInputSpec,
                                     runtime_profile)
 import pytest
+import sys
 
 run_profile = runtime_profile
 

nipype/pipeline/engine/tests/test_engine.py

@@ -10,7 +10,7 @@
 from builtins import open
 from copy import deepcopy
 from glob import glob
-import os
+import os, sys
 
 import networkx as nx
 

nipype/pipeline/plugins/tests/test_callback.py

@@ -7,7 +7,8 @@
 
 from builtins import object
 
-import pytest, pdb
+import pytest
+import sys
 import nipype.interfaces.utility as niu
 import nipype.pipeline.engine as pe