fix collator and cropping

809a9861 · Kolja Stahl · e9d2d893 · 809a9861
Commit 809a9861 authored Jan 13, 2022 by Kolja Stahl
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
@@ -505,7 +505,7 @@ class OpenFoldDataModule(pl.LightningDataModule):
        else:
            raise ValueError("Invalid stage")

-        batch_collator = OpenFoldBatchCollator(self.config, stage)
+        batch_collator = OpenFoldBatchCollator(self.config, generator, stage)

        dl = OpenFoldDataLoader(
            dataset,