diff --git a/torchgeo/datasets/landcoverai.py b/torchgeo/datasets/landcoverai.py index 3a5302e1e..a513a3407 100644 --- a/torchgeo/datasets/landcoverai.py +++ b/torchgeo/datasets/landcoverai.py @@ -74,6 +74,11 @@ class LandCoverAI(VisionDataset): transforms: a function/transform that takes input sample and its target as entry and returns a transformed version download: if True, download dataset and store it in the root directory + + Raises: + AssertionError: if ``split`` argument is invalid + RuntimeError: if ``download=False`` and data is not found, or checksums + don't match """ assert split in ["train", "val", "test"] diff --git a/torchgeo/datasets/nwpu.py b/torchgeo/datasets/nwpu.py index c66f9ac6d..b293114ee 100644 --- a/torchgeo/datasets/nwpu.py +++ b/torchgeo/datasets/nwpu.py @@ -93,6 +93,11 @@ class VHR10(VisionDataset): transforms: a function/transform that takes input sample and its target as entry and returns a transformed version download: if True, download dataset and store it in the root directory + + Raises: + AssertionError: if ``split`` argument is invalid + RuntimeError: if ``download=False`` and data is not found, or checksums + don't match """ assert split in ["positive", "negative"]