cleaning up imports

Author: oesteban

nipype/interfaces/base.py

@@ -9,11 +9,8 @@
 Requires Packages to be installed
 """
 from __future__ import print_function, division, unicode_literals, absolute_import
-from future import standard_library
-standard_library.install_aliases()
 from builtins import range, object, open, str, bytes
 
-from configparser import NoOptionError
 from copy import deepcopy
 import datetime
 from datetime import datetime as dt
@@ -26,7 +23,6 @@
 import select
 import subprocess as sp
 import sys
-import time
 from textwrap import wrap
 from warnings import warn
 import simplejson as json
@@ -44,6 +40,9 @@
     File, Directory, DictStrStr, has_metadata, ImageFile)
 from ..external.due import due
 
+from future import standard_library
+standard_library.install_aliases()
+
 nipype_version = Version(__version__)
 iflogger = logging.getLogger('interface')
 
@@ -55,11 +54,6 @@
 __docformat__ = 'restructuredtext'
 
 
-class Str(traits.Unicode):
-    """Replacement for the default traits.Str based in bytes"""
-
-traits.Str = Str
-
 
 class NipypeInterfaceError(Exception):
     """Custom error for interfaces"""

nipype/interfaces/traits_extension.py

@@ -18,23 +18,30 @@
 """
 from __future__ import print_function, division, unicode_literals, absolute_import
 
-from builtins import filter, object, str, bytes
+from builtins import str, bytes
 import os
 
 # perform all external trait imports here
-import traits
-if traits.__version__ < '3.7.0':
-    raise ImportError('Traits version 3.7.0 or higher must be installed')
+from traits import __version__ as traits_version
 import traits.api as traits
 from traits.trait_handlers import TraitDictObject, TraitListObject
 from traits.trait_errors import TraitError
 from traits.trait_base import _Undefined, class_of
 
 from traits.api import BaseUnicode
 from traits.api import Unicode
+if traits_version < '3.7.0':
+    raise ImportError('Traits version 3.7.0 or higher must be installed')
 
 DictStrStr = traits.Dict((bytes, str), (bytes, str))
-Str = Unicode
+
+
+class Str(traits.Unicode):
+    """Replacement for the default traits.Str based in bytes"""
+
+
+traits.Str = Str
+
 
 class BaseFile(BaseUnicode):
     """ Defines a trait whose value must be the name of a file.