diff --git a/torch_em/transform/augmentation.py b/torch_em/transform/augmentation.py index ae3ced13..f4dc6c74 100644 --- a/torch_em/transform/augmentation.py +++ b/torch_em/transform/augmentation.py @@ -64,7 +64,7 @@ def apply_transform(self, input, params): # TODO implement 'require_halo', and estimate the halo from the transformations # so that we can load a bigger block and cut it away class KorniaAugmentationPipeline(torch.nn.Module): - interpolatable_torch_tpyes = [torch.float16, torch.float32, torch.float64] + interpolatable_torch_types = [torch.float16, torch.float32, torch.float64] interpolatable_numpy_types = [np.dtype('float32'), np.dtype('float64')] def __init__(self, *kornia_augmentations, dtype=torch.float32):