copilot changed an if to if true so things were breaking! should be o…

…k now
alan-turing-institute · Mar 4, 2024 · 58bd6b7 · 58bd6b7
1 parent 19bb4b3
commit 58bd6b7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/caked/dataloader.py b/src/caked/dataloader.py
@@ -149,12 +149,12 @@ def process(self, paths: list[str], datatype: str):
             transforms.remove("normalise")
 
         gaussianblur = False
-        if True:
+        if "gaussianblur" in transforms:
             gaussianblur = True
             transforms.remove("gaussianblur")
 
         shiftmin = False
-        if True:
+        if "shiftmin" in transforms:
             shiftmin = True
             transforms.remove("shiftmin")
 

diff --git a/tests/test_disk_io.py b/tests/test_disk_io.py
@@ -331,7 +331,7 @@ def test_processing_after_load():
     assert not test_loader.dataset.normalise
     assert not test_loader.dataset.shiftmin
     assert not test_loader.dataset.gaussianblur
-    test_loader.transformations = TRANSFORM_ALL_RESCALE
+    test_loader.transformations = ["normalise", "gaussianblur", "shiftmin", "rescale=0"]
     pre_dataset = test_loader.dataset
     test_loader.load(datapath=TEST_DATA_MRC, datatype=DATATYPE_MRC)
     post_dataset = test_loader.dataset