Move conditional import.

Author: lohedges

src/BioSimSpace/FreeEnergy/_relative.py

@@ -48,9 +48,6 @@
 from .. import _is_notebook
 from .._Exceptions import MissingSoftwareError as _MissingSoftwareError
 
-if _is_notebook:
-    from IPython.display import FileLink as _FileLink
-
 # Check that the analyse_freenrg script exists.
 if _sys.platform != "win32":
     _analyse_freenrg = _os.path.join(_getBinDir(), "analyse_freenrg")
@@ -332,8 +329,6 @@ def getData(self, name="data", file_link=False, work_dir=None):
         import pathlib as _pathlib
         import zipfile as _zipfile
 
-        from IPython.display import FileLink as _FileLink
-
         from .._Utils import cd as _cd
 
         if self._work_dir is None:
@@ -377,6 +372,8 @@ def getData(self, name="data", file_link=False, work_dir=None):
         # Return a link to the archive.
         if _is_notebook:
             if file_link:
+                from IPython.display import FileLink as _FileLink
+
                 # Create a FileLink to the archive.
                 f_link = _FileLink(zipname)
 

src/BioSimSpace/Sandpit/Exscientia/FreeEnergy/_alchemical_free_energy.py

@@ -42,9 +42,6 @@
 from .. import _is_notebook
 from .._Exceptions import MissingSoftwareError as _MissingSoftwareError
 
-if _is_notebook:
-    from IPython.display import FileLink as _FileLink
-
 # Check that the analyse_freenrg script exists.
 if _sys.platform != "win32":
     _analyse_freenrg = _os.path.join(_getBinDir(), "analyse_freenrg")
@@ -441,8 +438,6 @@ def getData(self, name="data", file_link=False, work_dir=None):
         import zipfile as _zipfile
         from glob import glob as _glob
 
-        from IPython.display import FileLink as _FileLink
-
         from .._Utils import cd as _cd
 
         if self._work_dir is None:
@@ -485,6 +480,8 @@ def getData(self, name="data", file_link=False, work_dir=None):
         # Return a link to the archive.
         if _is_notebook:
             if file_link:
+                from IPython.display import FileLink as _FileLink
+
                 # Create a FileLink to the archive.
                 f_link = _FileLink(zipname)