fixes integration tests
@@ -84,7 +84,7 @@ class TestIntegrationSlidingWindow(DistTestCase):
)
output_image = nib.load(output_file).get_fdata()
np.testing.assert_allclose(np.sum(output_image), 33621)
- np.testing.assert_allclose(output_image.shape, (28, 25, 63, 1))
+ np.testing.assert_allclose(output_image.shape, (28, 25, 63))
if __name__ == "__main__":
