diff --git a/hrdae/conf b/hrdae/conf index 4315279..fe8be34 160000 --- a/hrdae/conf +++ b/hrdae/conf @@ -1 +1 @@ -Subproject commit 43152790a67359e11b2539baf56d710f7698df08 +Subproject commit fe8be34d8285e2880a3773274af6ca4c4e8f83fd diff --git a/hrdae/dataloaders/datasets/seq_divide_wrapper.py b/hrdae/dataloaders/datasets/seq_divide_wrapper.py index dd2706b..0fd7993 100644 --- a/hrdae/dataloaders/datasets/seq_divide_wrapper.py +++ b/hrdae/dataloaders/datasets/seq_divide_wrapper.py @@ -17,7 +17,7 @@ def __len__(self) -> int: return len(self.base) * self.period # type: ignore def __getitem__(self, index: int) -> dict[str, Tensor]: - x = self.base[index // self.period]["x+"] + x = self.base[index // self.period]["xp"] return { "x": x[index % self.period], "t": x[index % self.period],