fix: Improve error message clarity in CellMapImage and correct typo i…

…n CellMapMultiDataset
janelia-cellmap · Nov 21, 2024 · 4a57967 · 4a57967
1 parent 8bb1c6b
commit 4a57967
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/cellmap_data/image.py b/src/cellmap_data/image.py
@@ -362,9 +362,8 @@ def class_counts(self) -> float:
                 self._bg_count = bg_count * np.prod(list(self.scale.values()))
             except Exception as e:
                 # print(f"Error: {e}")
-                print(
-                    "Unable to get class counts from metadata, falling back to giving foreground 1 pixel, and the rest to background."
-                )
+                print(f"Unable to get class counts for {self.path}")
+                # print("from metadata, falling back to giving foreground 1 pixel, and the rest to background.")
                 self._class_counts = np.prod(list(self.scale.values()))
                 self._bg_count = (
                     np.prod(self.group[self.scale_level].shape) - 1

diff --git a/src/cellmap_data/multidataset.py b/src/cellmap_data/multidataset.py
@@ -157,7 +157,7 @@ def validation_indices(self) -> Sequence[int]:
                     if i == 0:
                         offset = 0
                     else:
-                        offset = self.cummulative_sizes[i - 1]
+                        offset = self.cumulative_sizes[i - 1]
                     sample_indices = np.array(dataset.validation_indices) + offset
                     indices.extend(list(sample_indices))
                 except AttributeError: