Skip to content

Commit

Permalink
fix typo in augmentations
Browse files Browse the repository at this point in the history
  • Loading branch information
JonasHell committed May 12, 2022
1 parent 7726828 commit 2e008e0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch_em/transform/augmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 2e008e0

Please sign in to comment.