diff --git a/tests/image_generation/helpers/image_generation_controlnet_test_helper.py b/tests/image_generation/helpers/image_generation_controlnet_test_helper.py index f89eb1f..d132fd1 100644 --- a/tests/image_generation/helpers/image_generation_controlnet_test_helper.py +++ b/tests/image_generation/helpers/image_generation_controlnet_test_helper.py @@ -48,7 +48,7 @@ class ImageGeneratorControlnetTestHelper: controlnet_strength=controlnet_strength, ), ) - image.save(path=output_image_path) + image.save(path=output_image_path, overwrite=True) # then np.testing.assert_array_equal( diff --git a/tests/image_generation/helpers/image_generation_test_helper.py b/tests/image_generation/helpers/image_generation_test_helper.py index 0ef548c..6718dc1 100644 --- a/tests/image_generation/helpers/image_generation_test_helper.py +++ b/tests/image_generation/helpers/image_generation_test_helper.py @@ -48,7 +48,7 @@ class ImageGeneratorTestHelper: width=width, ), ) - image.save(path=output_image_path) + image.save(path=output_image_path, overwrite=True) # then np.testing.assert_array_equal(