Allow disabling of lazy import via environment variable. [ci skip]

Author: lohedges

python/BioSimSpace/Sandpit/Exscientia/__init__.py

@@ -176,7 +176,6 @@ def _isVerbose():
         except:
             pass
 
-del _environ
 del _SireBase
 
 _gmx_path = None
@@ -226,29 +225,18 @@ def _isVerbose():
     del _shlex
     del _subprocess
 
-# Don't lazy load the submodules if not in a notebook.
-if _is_notebook:
-    from . import Align
-    from . import Box
-    from . import Convert
-    from . import FreeEnergy
-    from . import Gateway
-    from . import IO
-    from . import Metadynamics
-    from . import MD
-    from . import Node
-    from . import Notebook
-    from . import Parameters
-    from . import Process
-    from . import Protocol
-    from . import Solvent
-    from . import Stream
-    from . import Trajectory
-    from . import Types
-    from . import Units
-
+# Whether to lazy load submodules.
+if (
+    "SIRE_NO_LAZY_IMPORT" in _environ
+    or "BSS_NO_LAZY_IMPORT" in _environ
+    or _is_notebook
+):
+    _can_lazy_import = False
 else:
-    # Import the lazy import module.
+    _can_lazy_import = True
+
+# Lazy load submodules if possible.
+if _can_lazy_import:
     import lazy_import as _lazy_import
 
     Align = _lazy_import.lazy_module("BioSimSpace.Sandpit.Exscientia.Align")
@@ -272,7 +260,29 @@ def _isVerbose():
     Types = _lazy_import.lazy_module("BioSimSpace.Sandpit.Exscientia.Types")
     Units = _lazy_import.lazy_module("BioSimSpace.Sandpit.Exscientia.Units")
 
-del _lazy_import
+    del _lazy_import
+else:
+    from . import Align
+    from . import Box
+    from . import Convert
+    from . import FreeEnergy
+    from . import Gateway
+    from . import IO
+    from . import Metadynamics
+    from . import MD
+    from . import Node
+    from . import Notebook
+    from . import Parameters
+    from . import Process
+    from . import Protocol
+    from . import Solvent
+    from . import Stream
+    from . import Trajectory
+    from . import Types
+    from . import Units
+
+del _can_lazy_import
+del _environ
 
 # Import Versioneer from the package root.
 from ... import _version

python/BioSimSpace/__init__.py

@@ -176,7 +176,6 @@ def _isVerbose():
         except:
             pass
 
-del _environ
 del _SireBase
 
 _gmx_path = None
@@ -226,28 +225,18 @@ def _isVerbose():
     del _shlex
     del _subprocess
 
-# Don't lazy load submodules if in a notebook environment.
-if _is_notebook:
-    from . import Align
-    from . import Box
-    from . import Convert
-    from . import FreeEnergy
-    from . import Gateway
-    from . import IO
-    from . import Metadynamics
-    from . import MD
-    from . import Node
-    from . import Notebook
-    from . import Parameters
-    from . import Process
-    from . import Protocol
-    from . import Solvent
-    from . import Stream
-    from . import Trajectory
-    from . import Types
-    from . import Units
-
+# Whether to lazy load submodules.
+if (
+    "SIRE_NO_LAZY_IMPORT" in _environ
+    or "BSS_NO_LAZY_IMPORT" in _environ
+    or _is_notebook
+):
+    _can_lazy_import = False
 else:
+    _can_lazy_import = True
+
+# Lazy import submodules if possible.
+if _can_lazy_import:
     import lazy_import as _lazy_import
 
     Align = _lazy_import.lazy_module("BioSimSpace.Align")
@@ -270,6 +259,28 @@ def _isVerbose():
     Units = _lazy_import.lazy_module("BioSimSpace.Units")
 
     del _lazy_import
+else:
+    from . import Align
+    from . import Box
+    from . import Convert
+    from . import FreeEnergy
+    from . import Gateway
+    from . import IO
+    from . import Metadynamics
+    from . import MD
+    from . import Node
+    from . import Notebook
+    from . import Parameters
+    from . import Process
+    from . import Protocol
+    from . import Solvent
+    from . import Stream
+    from . import Trajectory
+    from . import Types
+    from . import Units
+
+del _can_lazy_import
+del _environ
 
 from . import _version