make the new extensions accessible

kushalkolar · Jul 14, 2021 · 9a4e5a7 · 9a4e5a7
1 parent 269778c
commit 9a4e5a7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/mesmerize/analysis/__init__.py b/mesmerize/analysis/__init__.py
@@ -3,3 +3,4 @@
 from .math import cross_correlation, drfft_dtw, tvregdiff
 from . import clustering_metrics
 from . import meta_clustering
+from .batch_manager_reader import load_batch, CaimanExtensions
diff --git a/mesmerize/analysis/batch_manager_reader.py b/mesmerize/analysis/batch_manager_reader.py
@@ -29,7 +29,7 @@ def _get_item_uuid(item: Union[int, str, UUID]) -> UUID:
 
 
 @pd.api.extensions.register_series_accessor("caiman")
-class Caiman:
+class CaimanExtensions:
     """
     Extensions for caiman related functions
     """