ci(generative_ai): skip tests for imagegeneration@00x, imagetext@001 samples (#13615)

* ci(generative_ai): skip tests for imagegeneration@006 samples

b/452455170

* lint

* merge #13608 skips, to solve snippets-scoped tests

* skip imagegeneration@002 samples (deprecated model)

Author: glasnt

generative_ai/image_generation/edit_image_inpainting_insert_mask_mode_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_inpainting_insert_mask_mode
 
@@ -28,6 +29,7 @@
 _PROMPT = "beach"
 
 
+@pytest.mark.skip("imagegeneration@006 samples pending deprecation")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_edit_image_inpainting_insert_mask_mode() -> None:
     response = (

generative_ai/image_generation/edit_image_inpainting_insert_mask_test.py

@@ -16,6 +16,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_inpainting_insert_mask
 
@@ -27,6 +28,7 @@
 _PROMPT = "hat"
 
 
+@pytest.mark.skip("imagegeneration@006 samples pending deprecation")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_edit_image_inpainting_insert_mask() -> None:
     response = edit_image_inpainting_insert_mask.edit_image_inpainting_insert_mask(

generative_ai/image_generation/edit_image_inpainting_remove_mask_mode_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_inpainting_remove_mask_mode
 
@@ -28,6 +29,7 @@
 _PROMPT = "sports car"
 
 
+@pytest.mark.skip("imagegeneration@006 samples pending deprecation")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_edit_image_inpainting_remove_mask_mode() -> None:
     response = (

generative_ai/image_generation/edit_image_inpainting_remove_mask_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_inpainting_remove_mask
 
@@ -28,6 +29,7 @@
 _PROMPT = "volleyball game"
 
 
+@pytest.mark.skip("imagegeneration@006 samples pending deprecation")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_edit_image_inpainting_remove_mask() -> None:
     response = edit_image_inpainting_remove_mask.edit_image_inpainting_remove_mask(

generative_ai/image_generation/edit_image_mask_free_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_mask_free
 
@@ -27,6 +28,7 @@
 _PROMPT = "a dog"
 
 
+@pytest.mark.skip("imagegeneration@002 samples pending deprecation")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_edit_image_mask_free() -> None:
     response = edit_image_mask_free.edit_image_mask_free(

generative_ai/image_generation/edit_image_mask_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_mask
 
@@ -28,6 +29,7 @@
 _PROMPT = "a big book"
 
 
+@pytest.mark.skip("imagegeneration@002 samples pending deprecation")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_edit_image_mask() -> None:
     response = edit_image_mask.edit_image_mask(

generative_ai/image_generation/edit_image_outpainting_mask_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_outpainting_mask
 
@@ -28,6 +29,7 @@
 _PROMPT = "city with skyscrapers"
 
 
+@pytest.mark.skip("imagegeneration@006 samples pending deprecation")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_edit_image_outpainting_mask() -> None:
     response = edit_image_outpainting_mask.edit_image_outpainting_mask(

generative_ai/image_generation/edit_image_product_image_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_product_image
 
@@ -27,6 +28,7 @@
 _PROMPT = "beach"
 
 
+@pytest.mark.skip("imagegeneration@006 samples pending deprecation")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_edit_image_product_image() -> None:
     response = edit_image_product_image.edit_image_product_image(

generative_ai/image_generation/get_short_form_image_captions_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import get_short_form_image_captions
 
@@ -25,6 +26,7 @@
 _INPUT_FILE = os.path.join(_RESOURCES, "cat.png")
 
 
+@pytest.mark.skip("Sample pending deprecation b/452720552")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_get_short_form_image_captions() -> None:
     response = get_short_form_image_captions.get_short_form_image_captions(

generative_ai/image_generation/get_short_form_image_responses_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import get_short_form_image_responses
 
@@ -26,6 +27,7 @@
 _QUESTION = "What breed of cat is this a picture of?"
 
 
+@pytest.mark.skip("Sample pending deprecation b/452720552")
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=60)
 def test_get_short_form_image_responses() -> None:
     response = get_short_form_image_responses.get_short_form_image_responses(