Skip to content

Commit 6ede899

Browse files
committed
chore(genai): remove migrated and unused image generation samples
Deletes unused image generation examples (subject/control reference, image upscaling, and basic text-to-image) and updates the corresponding tests.
1 parent 49a9123 commit 6ede899

4 files changed

Lines changed: 0 additions & 188 deletions

File tree

genai/image_generation/imggen_subj_refer_ctrl_refer_with_txt_imgs.py

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

genai/image_generation/imggen_upscale_with_img.py

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

genai/image_generation/imggen_with_txt.py

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

genai/image_generation/test_image_generation.py

Lines changed: 0 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,7 @@
3636
import imggen_raw_reference_with_txt_img
3737
import imggen_scribble_ctrl_type_with_txt_img
3838
import imggen_style_reference_with_txt_img
39-
import imggen_subj_refer_ctrl_refer_with_txt_imgs
40-
import imggen_upscale_with_img
4139
import imggen_virtual_try_on_with_txt_img
42-
import imggen_with_txt
4340

4441
os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
4542
os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
@@ -63,12 +60,6 @@ def output_gcs_uri() -> str:
6360
blob.delete()
6461

6562

66-
def test_img_generation() -> None:
67-
OUTPUT_FILE = os.path.join(RESOURCES, "dog_newspaper.png")
68-
response = imggen_with_txt.generate_images(OUTPUT_FILE)
69-
assert response
70-
71-
7263
def test_img_edit_inpainting_insert_with_mask() -> None:
7364
OUTPUT_FILE = os.path.join(RESOURCES, "fruit_edit.png")
7465
response = imggen_inpainting_insert_mask_with_txt_img.edit_inpainting_insert_mask(OUTPUT_FILE)
@@ -117,13 +108,6 @@ def test_img_edit_mask_free() -> None:
117108
assert response
118109

119110

120-
def test_img_customization_subject(output_gcs_uri: str) -> None:
121-
response = imggen_subj_refer_ctrl_refer_with_txt_imgs.subject_customization(
122-
output_gcs_uri=output_gcs_uri
123-
)
124-
assert response
125-
126-
127111
def test_img_customization_style(output_gcs_uri: str) -> None:
128112
response = imggen_style_reference_with_txt_img.style_customization(output_gcs_uri=output_gcs_uri)
129113
assert response
@@ -148,9 +132,3 @@ def test_img_virtual_try_on() -> None:
148132
OUTPUT_FILE = os.path.join(RESOURCES, "man_in_sweater.png")
149133
response = imggen_virtual_try_on_with_txt_img.virtual_try_on(OUTPUT_FILE)
150134
assert response
151-
152-
153-
def test_img_upscale() -> None:
154-
OUTPUT_FILE = os.path.join(RESOURCES, "dog_newspaper.png")
155-
response = imggen_upscale_with_img.upscale_images(OUTPUT_FILE)
156-
assert response

0 commit comments

Comments
 (0)