add docstrings

catalystneuro · Mar 4, 2024 · 1e2ce74 · 1e2ce74
1 parent 3ee2693
commit 1e2ce74
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/roiextractors/extractors/caiman/caimansegmentationextractor.py b/src/roiextractors/extractors/caiman/caimansegmentationextractor.py
@@ -114,10 +114,12 @@ def _trace_extractor_read(self, field):
             return lazy_ops.DatasetView(self._dataset_file["estimates"][field]).lazy_transpose()
 
     def _correlation_image_read(self):
+        """Read correlation image Cn."""
         if self._dataset_file["estimates"].get("Cn"):
             return np.array(self._dataset_file["estimates"]["Cn"])
 
     def _summary_image_read(self):
+        """Read summary image mean."""
         if self._dataset_file["estimates"].get("b"):
             FOV_shape = self._dataset_file["params"]["data"]["dims"][()]
             b_sum = self._dataset_file["estimates"]["b"][:].sum(axis=1)