Start local import for neo.rawio

Author: samuelgarcia

neo/rawio/baserawio.py

@@ -75,12 +75,6 @@
 
 from neo import logging_handler
 
-try:
-    import joblib
-
-    HAVE_JOBLIB = True
-except ImportError:
-    HAVE_JOBLIB = False
 
 possible_raw_modes = ['one-file', 'multi-file', 'one-dir', ]  # 'multi-dir', 'url', 'other'
 
@@ -162,7 +156,7 @@ def __init__(self, use_cache=False, cache_path='same_as_resource', **kargs):
 
         self.use_cache = use_cache
         if use_cache:
-            assert HAVE_JOBLIB, 'You need to install joblib for cache'
+            import joblib
             self.setup_cache(cache_path)
         else:
             self._cache = None
@@ -703,6 +697,8 @@ def rescale_epoch_duration(self, raw_duration, dtype='float64',
         return self._rescale_epoch_duration(raw_duration, dtype, event_channel_index)
 
     def setup_cache(self, cache_path, **init_kargs):
+        import joblib
+
         if self.rawmode in ('one-file', 'multi-file'):
             resource_name = self.filename
         elif self.rawmode == 'one-dir':

neo/rawio/biocamrawio.py

@@ -12,11 +12,6 @@
 
 import numpy as np
 
-try:
-    import h5py
-    HAVE_H5PY = True
-except ImportError:
-    HAVE_H5PY = False
 
 
 class BiocamRawIO(BaseRawIO):
@@ -45,7 +40,6 @@ def _source_name(self):
         return self.filename
 
     def _parse_header(self):
-        assert HAVE_H5PY, 'h5py is not installed'
         self._header_dict = open_biocam_file_header(self.filename)
         self._num_channels = self._header_dict["num_channels"]
         self._num_frames = self._header_dict["num_frames"]
@@ -121,7 +115,7 @@ def _get_analogsignal_chunk(self, block_index, seg_index, i_start, i_stop,
 def open_biocam_file_header(filename):
     """Open a Biocam hdf5 file, read and return the recording info, pick the correct method to access raw data,
     and return this to the caller."""
-    assert HAVE_H5PY, 'h5py is not installed'
+    import h5py
 
     rf = h5py.File(filename, 'r')
     # Read recording variables

neo/rawio/cedrawio.py

@@ -26,11 +26,6 @@
 import numpy as np
 from copy import deepcopy
 
-try:
-    import sonpy
-    HAVE_SONPY = True
-except ImportError:
-    HAVE_SONPY = False
 
 
 class CedRawIO(BaseRawIO):
@@ -53,7 +48,7 @@ def _source_name(self):
         return self.filename
 
     def _parse_header(self):
-        assert HAVE_SONPY, 'sonpy must be installed'
+        import sonpy
 
         self.smrx_file = sonpy.lib.SonFile(sName=str(self.filename), bReadOnly=True)
         smrx = self.smrx_file

neo/rawio/maxwellrawio.py

@@ -27,12 +27,6 @@
 
 import numpy as np
 
-try:
-    import h5py
-    HAVE_H5 = True
-except ImportError:
-    HAVE_H5 = False
-
 
 class MaxwellRawIO(BaseRawIO):
     """
@@ -50,12 +44,7 @@ def _source_name(self):
         return self.filename
 
     def _parse_header(self):
-        try:
-            import MEArec as mr
-            HAVE_MEAREC = True
-        except ImportError:
-            HAVE_MEAREC = False
-        assert HAVE_H5, 'h5py is not installed'
+        import h5py
 
         h5 = h5py.File(self.filename, mode='r')
         self.h5_file = h5

neo/rawio/mearecrawio.py

@@ -44,12 +44,7 @@ def _source_name(self):
         return self.filename
 
     def _parse_header(self):
-        try:
-            import MEArec as mr
-            HAVE_MEAREC = True
-        except ImportError:
-            HAVE_MEAREC = False
-        assert HAVE_MEAREC, 'MEArec is not installed'
+        import MEArec as mr
         self._recgen = mr.load_recordings(recordings=self.filename, return_h5_objects=True,
                                           check_suffix=False,
                                           load=['recordings', 'spiketrains', 'channel_positions'],

neo/rawio/nixrawio.py

@@ -15,13 +15,6 @@
                         _spike_channel_dtype, _event_channel_dtype)
 from ..io.nixio import check_nix_version
 
-try:
-    import nixio as nix
-
-    HAVE_NIX = True
-except ImportError:
-    HAVE_NIX = False
-    nix = None
 
 
 # When reading metadata properties, the following keys are ignored since they
@@ -50,6 +43,8 @@ def _source_name(self):
         return self.filename
 
     def _parse_header(self):
+        import nixio as nix
+
         self.file = nix.File.open(self.filename, nix.FileMode.ReadOnly)
         signal_channels = []
         anasig_ids = {0: []}  # ids of analogsignals by segment