From dd2452360df8e3ac51efce9768e4edbf334431de Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniel=20Z=C3=BCgner?= Date: Wed, 20 Apr 2022 10:01:38 +0200 Subject: [PATCH] Update datamodule.py Fix bug in lattice scaler assignment of test datasets. --- cdvae/pl_data/datamodule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cdvae/pl_data/datamodule.py b/cdvae/pl_data/datamodule.py index cd9a4380..36f296d5 100644 --- 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: