Skip to content

Commit 2ef6698

Browse files
authored
chore(genai): remove unused image generation samples 2/5 (#14554)
* chore(genai): remove unused image generation samples Deletes unused image generation examples (inpainting removal and mask-free edit) and updates the corresponding tests. * Fix import order for linter.
1 parent 123b291 commit 2ef6698

4 files changed

Lines changed: 2 additions & 207 deletions

File tree

‎genai/image_generation/imggen_inpainting_removal_mask_with_txt_img.py‎

Lines changed: 0 additions & 66 deletions
This file was deleted.

‎genai/image_generation/imggen_inpainting_removal_with_txt_img.py‎

Lines changed: 0 additions & 65 deletions
This file was deleted.

‎genai/image_generation/imggen_mask_free_edit_with_txt_img.py‎

Lines changed: 0 additions & 53 deletions
This file was deleted.

‎genai/image_generation/test_image_generation.py‎

Lines changed: 2 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,9 @@
2222

2323
from google.cloud import storage
2424

25-
import pytest
26-
2725
import imggen_canny_ctrl_type_with_txt_img
2826
import imggen_inpainting_insert_mask_with_txt_img
2927
import imggen_inpainting_insert_with_txt_img
30-
import imggen_inpainting_removal_mask_with_txt_img
31-
import imggen_inpainting_removal_with_txt_img
32-
import imggen_mask_free_edit_with_txt_img
3328
import imggen_outpainting_with_txt_img
3429
import imggen_product_background_mask_with_txt_img
3530
import imggen_product_background_with_txt_img
@@ -41,6 +36,8 @@
4136
import imggen_virtual_try_on_with_txt_img
4237
import imggen_with_txt
4338

39+
import pytest
40+
4441
os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
4542
os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
4643
# The project name is included in the CICD pipeline
@@ -81,18 +78,6 @@ def test_img_edit_inpainting_insert() -> None:
8178
assert response
8279

8380

84-
def test_img_edit_inpainting_removal_mask() -> None:
85-
OUTPUT_FILE = os.path.join(RESOURCES, "fruit_edit.png")
86-
response = imggen_inpainting_removal_mask_with_txt_img.edit_inpainting_removal_mask(OUTPUT_FILE)
87-
assert response
88-
89-
90-
def test_img_edit_inpainting_removal() -> None:
91-
OUTPUT_FILE = os.path.join(RESOURCES, "fruit_edit.png")
92-
response = imggen_inpainting_removal_with_txt_img.edit_inpainting_removal(OUTPUT_FILE)
93-
assert response
94-
95-
9681
def test_img_edit_product_background_mask() -> None:
9782
OUTPUT_FILE = os.path.join(RESOURCES, "suitcase_edit.png")
9883
response = imggen_product_background_mask_with_txt_img.edit_product_background_mask(OUTPUT_FILE)
@@ -111,12 +96,6 @@ def test_img_edit_outpainting() -> None:
11196
assert response
11297

11398

114-
def test_img_edit_mask_free() -> None:
115-
OUTPUT_FILE = os.path.join(RESOURCES, "latte_edit.png")
116-
response = imggen_mask_free_edit_with_txt_img.edit_mask_free(OUTPUT_FILE)
117-
assert response
118-
119-
12099
def test_img_customization_subject(output_gcs_uri: str) -> None:
121100
response = imggen_subj_refer_ctrl_refer_with_txt_imgs.subject_customization(
122101
output_gcs_uri=output_gcs_uri

0 commit comments

Comments
 (0)