Fix missing raw_sample_filter definition

facebookresearch · Aug 25, 2022 · 37ffc89 · 37ffc89
1 parent 6aefc03
commit 37ffc89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastmri/pl_modules/data_module.py b/fastmri/pl_modules/data_module.py
@@ -221,6 +221,7 @@ def _create_data_loader(
                     if volume_sample_rate is None
                     else volume_sample_rate
                 )
+                raw_sample_filter = self.test_filter
 
         # if desired, combine train and val together for the train split
         dataset: Union[SliceDataset, CombinedSliceDataset]
@@ -248,7 +249,6 @@ def _create_data_loader(
         else:
             if data_partition in ("test", "challenge") and self.test_path is not None:
                 data_path = self.test_path
-                raw_sample_filter = self.test_filter
             else:
                 data_path = self.data_path / f"{self.challenge}_{data_partition}"