Merge pull request #9 from danielzuegner/main

Fix bug in lattice scaler assignment of test datasets.
txie-93 · Apr 20, 2022 · f857f59 · f857f59
2 parents 76960ae + dd24523
commit f857f59
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cdvae/pl_data/datamodule.py b/cdvae/pl_data/datamodule.py
@@ -94,7 +94,7 @@ def setup(self, stage: Optional[str] = None):
                 for dataset_cfg in self.datasets.test
             ]
             for test_dataset in self.test_datasets:
-                test_dataset.lattice_scaler = self.scaler
+                test_dataset.lattice_scaler = self.lattice_scaler
                 test_dataset.scaler = self.scaler
 
     def train_dataloader(self) -> DataLoader: