diff --git a/httomo/transform_layer.py b/httomo/transform_layer.py index 8c25cdecc..956659a91 100644 --- a/httomo/transform_layer.py +++ b/httomo/transform_layer.py @@ -57,7 +57,7 @@ def insert_save_methods(self, pipeline: Pipeline) -> Pipeline: methods.append(m) if ( (m.save_result or self._save_all) - and m.method_name != "save_to_images" + and m.method_name not in ["save_to_images", "data_checker"] and "center" not in m.method_name ): methods.append( diff --git a/tests/test_transform_layer.py b/tests/test_transform_layer.py index 46554c99c..afc812921 100644 --- a/tests/test_transform_layer.py +++ b/tests/test_transform_layer.py @@ -54,8 +54,8 @@ def test_insert_save_methods_save_all(mocker: MockerFixture, tmp_path: Path): assert pipeline[3].task_id == "save_t2" -@pytest.mark.parametrize("method_name", ["save_to_images", "find_center_vo"]) -def test_insert_save_methods_does_not_save_for_save_to_images( +@pytest.mark.parametrize("method_name", ["save_to_images", "find_center_vo", "data_checker"]) +def test_insert_save_methods_does_not_save_for_some_methods( mocker: MockerFixture, method_name: str, tmp_path: Path ): comm = MPI.COMM_SELF