Skip to content

Commit

Permalink
fix(datamodule): typo self.augs instead of self.aug
Browse files Browse the repository at this point in the history
  • Loading branch information
Mathias Baumgartinger committed Dec 7, 2024
1 parent f6c52eb commit 61e2f2f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchgeo/datamodules/flair2.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ def __init__(
K.Normalize(mean=self.mean, std=self.std), data_keys=['image', 'mask']
)

self.augs = augs if augs is not None else self.aug
self.aug = augs if augs is not None else self.aug

def setup(self, stage: str) -> None:
"""Set up datasets.
Expand Down

0 comments on commit 61e2f2f

Please sign in to comment.