def _get_dataset(self, basin: str) -> BaseDataset:
"""Get dataset for a single basin."""
ds = get_dataset(cfg=self.cfg,
is_train=False,
period=self.period,
basin=basin,
additional_features=self.additional_features,
id_to_int=self.id_to_int,
scaler=self.scaler)
return ds
def _get_dataset(self) -> BaseDataset:
return get_dataset(cfg=self.cfg,
period="train",
is_train=True,
scaler=self._scaler)