sty: better placement of imports from enthought.traits

Author: oesteban

nipype/interfaces/base/__init__.py

@@ -8,6 +8,9 @@
 This module defines the API of all nipype interfaces.
 
 """
+from traits.trait_handlers import TraitDictObject, TraitListObject
+from traits.trait_errors import TraitError
+
 from .core import (Interface, BaseInterface, SimpleInterface, CommandLine,
                    StdOutCommandLine, MpiCommandLine, SEMLikeCommandLine,
                    LibraryBaseInterface, PackageInfo)
@@ -17,7 +20,7 @@
                     StdOutCommandLineInputSpec)
 
 from .traits_extension import (
-    traits, Undefined, TraitDictObject, TraitListObject, TraitError, isdefined,
+    traits, Undefined, isdefined,
     File, Directory, Str, DictStrStr, has_metadata, ImageFile,
     OutputMultiObject, InputMultiObject,
     OutputMultiPath, InputMultiPath)

nipype/interfaces/base/core.py

@@ -27,6 +27,7 @@
 import simplejson as json
 from dateutil.parser import parse as parseutc
 from future import standard_library
+from traits.trait_errors import TraitError
 
 from ... import config, logging, LooseVersion
 from ...utils.provenance import write_provenance
@@ -37,7 +38,7 @@
 
 from ...external.due import due
 
-from .traits_extension import traits, isdefined, TraitError
+from .traits_extension import traits, isdefined
 from .specs import (BaseInterfaceInputSpec, CommandLineInputSpec,
                     StdOutCommandLineInputSpec, MpiCommandLineInputSpec,
                     get_filecopy_info)