import datatlad only when running download_dataset()

Author: samuelgarcia

neo/utils/__init__.py

@@ -5,5 +5,4 @@
 from .misc import (get_events, get_epochs, add_epoch,
     match_events, cut_block_by_epochs, cut_segment_by_epoch,
     is_block_rawio_compatible)
-from .datasets import (download_dataset,
-    get_local_testing_data_folder, HAVE_DATALAD)
+from .datasets import download_dataset, get_local_testing_data_folder

neo/utils/datasets.py

@@ -4,14 +4,6 @@
 import os
 from pathlib import Path
 
-try:
-    import datalad.api
-    from datalad.support.gitrepo import GitRepo
-
-    HAVE_DATALAD = True
-except:
-    HAVE_DATALAD = False
-
 default_testing_repo = 'https://gin.g-node.org/NeuralEnsemble/ephy_testing_data'
 
 global local_testing_data_folder
@@ -59,7 +51,8 @@ def download_dataset(repo=default_testing_repo, remote_path=None,
     local_path:
         The local path of the downloaded file or folder
     """
-    assert HAVE_DATALAD, 'You need to install datalad'
+    import datalad.api
+    from datalad.support.gitrepo import GitRepo
 
     if local_folder is None:
         global local_testing_data_folder