conditional import of caiman

kushalkolar · Sep 16, 2021 · 2e77eab · 2e77eab
1 parent 47d1c53
commit 2e77eab
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/mesmerize/analysis/batch_manager_reader.py b/mesmerize/analysis/batch_manager_reader.py
@@ -4,10 +4,12 @@
 
 import pandas as pd
 import pathlib
-import caiman as cm
+from ..common import configuration
 from typing import *
 from uuid import UUID
 import pickle
+if configuration.HAS_CAIMAN:
+    import caiman as cm
 
 CURRENT_BATCH: pathlib.Path = None  # only one batch at a time for now
 
@@ -34,6 +36,9 @@ class CaimanExtensions:
     Extensions for caiman related functions
     """
     def __init__(self, series: pd.Series):
+        if not configuration.HAS_CAIMAN:
+            raise ImportError("Caiman not found in the environment")
+
         self._series = series
 
     def get_params(self) -> dict: