diff --git a/kwant/__init__.py b/kwant/__init__.py index 1fddab25426261753a81e358a30ddab009ac3171..054657c308607608b1c9c8a893680231b21c59f6 100644 --- a/kwant/__init__.py +++ b/kwant/__init__.py @@ -33,7 +33,7 @@ from ._common import KwantDeprecationWarning, UserCodeError __all__.extend(['KwantDeprecationWarning', 'UserCodeError']) for module in ['system', 'builder', 'lattice', 'solvers', 'digest', 'rmt', - 'operator', 'kpm', 'wraparound', 'continuum']: + 'operator', 'kpm', 'wraparound']: exec('from . import {0}'.format(module)) __all__.append(module) @@ -56,6 +56,13 @@ except: else: __all__.extend(['plotter', 'plot']) +# Lazy import continuum package for backwards compatibility +from ._common import lazy_import + +continuum = lazy_import('continuum', deprecation_warning=True) +__all__.append('continuum') +del lazy_import + def test(verbose=True): from pytest import main