Skip to content

Commit

Permalink
Merge pull request #59 from JonasHell/main
Browse files Browse the repository at this point in the history
fix typo in augmentations
  • Loading branch information
constantinpape committed May 12, 2022
2 parents 7726828 + 2e008e0 commit 83c8d0f
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 83c8d0f

Please sign in to comment.