Skip to content

Commit

Permalink
update kwargs
Browse files Browse the repository at this point in the history
  • Loading branch information
yichiac committed Feb 17, 2024
1 parent 3107838 commit 0b77b32
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torchgeo/datamodules/cdlsentinel2.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,11 +40,11 @@ def __init__(
**kwargs: Additional keyword arguments passed to
:class:`~torchgeo.datasets.CDL`.
"""
self.nccm_kwargs = {}
self.cdl_kwargs = {}
self.sentinel2_kwargs = {}
for key, val in kwargs.items():
if key.startswith("cdl_"):
self.nccm_kwargs[key[5:]] = val
self.cdl_kwargs[key[5:]] = val
elif key.startswith("sentinel2_"):
self.sentinel2_kwargs[key[10:]] = val

Expand Down

0 comments on commit 0b77b32

Please sign in to comment.