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

b/452455170

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
 
@@ -27,7 +28,7 @@
 _OUTPUT_FILE = os.path.join(_RESOURCES, "woman_at_beach.png")
 _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
 
@@ -26,7 +27,7 @@
 _OUTPUT_FILE = os.path.join(_RESOURCES, "woman_with_hat.png")
 _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
 
@@ -27,7 +28,7 @@
 _OUTPUT_FILE = os.path.join(_RESOURCES, "sports_car.png")
 _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
 
@@ -27,7 +28,7 @@
 _OUTPUT_FILE = os.path.join(_RESOURCES, "volleyball_game_single_blue_player.png")
 _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_outpainting_mask_test.py

@@ -17,6 +17,7 @@
 import backoff
 
 from google.api_core.exceptions import ResourceExhausted
+import pytest
 
 import edit_image_outpainting_mask
 
@@ -27,7 +28,7 @@
 _OUTPUT_FILE = os.path.join(_RESOURCES, "roller_skaters_downtown.png")
 _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
 
@@ -26,7 +27,7 @@
 _OUTPUT_FILE = os.path.join(_RESOURCES, "pillow_on_beach.png")
 _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(