fix seq_divide_wrapper

nnaakkaaii · Jun 22, 2024 · bb447de · bb447de
1 parent 53594c2
commit bb447de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/hrdae/conf b/hrdae/conf
diff --git 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],