diff --git a/qim3d/utils/augmentations.py b/qim3d/utils/augmentations.py index ed93955ffbdd91cb02219d617c492c3aa14752cf..aee677798b0bada9e8a9ad541673a64016cc499e 100644 --- a/qim3d/utils/augmentations.py +++ b/qim3d/utils/augmentations.py @@ -31,7 +31,7 @@ class Augmentation: ): if resize not in ['crop', 'reshape', 'padding']: - raise ValueError(f"Invalid resize type: {resize}. Use either 'crop', 'resize' or 'padding'.") + raise ValueError(f"Invalid resize type: {resize}. Use either 'crop', 'reshape' or 'padding'.") self.resize = resize self.mean = mean