diff --git a/avae/data.py b/avae/data.py index 993fcc0..1e86e78 100644 --- a/avae/data.py +++ b/avae/data.py @@ -349,8 +349,15 @@ def read(self, filename): return np.load(os.path.join(self.root_dir, filename)) elif self.datatype == "mrc": - with mrcfile.open(os.path.join(self.root_dir, filename)) as f: - return np.array(f.data) + try: + with mrcfile.open(os.path.join(self.root_dir, filename)) as f: + return np.array(f.data) + except ValueError: + raise ValueError( + "File {} is corrupted.".format( + os.path.join(self.root_dir, filename) + ) + ) def voxel_transformation(self, x):