From 3e80ca8459864052fc04a81e77112f719baee33e Mon Sep 17 00:00:00 2001 From: Dingel321 Date: Tue, 25 Jun 2024 17:12:23 +0200 Subject: [PATCH] fixed bug in mrc_dataset --- src/cryo_sbi/utils/image_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cryo_sbi/utils/image_utils.py b/src/cryo_sbi/utils/image_utils.py index 4bab630..8a94ad5 100644 --- a/src/cryo_sbi/utils/image_utils.py +++ b/src/cryo_sbi/utils/image_utils.py @@ -389,7 +389,7 @@ def load_index_map(self, path: str): path (str): Path to load the index map. """ index_map = np.load(path) - assert self.paths == index_map["paths"], "Paths do not match the index map." + assert all(self.paths == index_map["paths"]), "Paths do not match the index map." self._path_index = index_map["path_index"] self._file_index = index_map["file_index"] self._index_map = True