Skip to content

Commit

Permalink
add versionchanged and versionadded
Browse files Browse the repository at this point in the history
  • Loading branch information
isaaccorley committed Apr 26, 2023
1 parent 38c6a04 commit 93c54a3
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 0 deletions.
5 changes: 5 additions & 0 deletions torchgeo/datamodules/fair1m.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,8 @@ def collate_fn(batch: list[dict[str, Tensor]]) -> dict[str, Any]:
Returns:
batch dict output
.. versionadded:: 0.5
"""
output: dict[str, Any] = {}
output["image"] = torch.stack([sample["image"] for sample in batch])
Expand Down Expand Up @@ -48,6 +50,9 @@ def __init__(
num_workers: Number of workers for parallel data loading.
**kwargs: Additional keyword arguments passed to
:class:`~torchgeo.datasets.FAIR1M`.
.. versionchanged:: 0.5
Removed *val_split_pct* and *test_split_pct* parameters.
"""
super().__init__(FAIR1M, batch_size, num_workers, **kwargs)
self.collate_fn = collate_fn
Expand Down
3 changes: 3 additions & 0 deletions torchgeo/datasets/fair1m.py
Original file line number Diff line number Diff line change
Expand Up @@ -245,6 +245,9 @@ def __init__(
AssertionError: if ``split`` argument is invalid
RuntimeError: if ``download=False`` and data is not found, or checksums
don't match
.. versionchanged:: 0.5
Added *split* and *download* parameters.
"""
assert split in self.directories
self.root = root
Expand Down

0 comments on commit 93c54a3

Please sign in to comment.