Fix bug in dummy dataset

428d90ce · Gustaf Ahdritz · 94ab346e · 428d90ce
Commit 428d90ce authored Oct 25, 2021 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openfold/data/data_modules.py openfold/data/data_modules.py +1 -1

No files found.
--- a/openfold/data/data_modules.py
+++ b/openfold/data/data_modules.py
@@ -453,7 +453,7 @@ class OpenFoldDataModule(pl.LightningDataModule):
 class DummyDataset(torch.utils.data.Dataset):
    def __init__(self, batch_path):
        with open(batch_path, "rb") as f:
-            batch = pickle.load(f)
+            self.batch = pickle.load(f)

    def __getitem__(self, idx):
        return copy.deepcopy(self.batch)