skip imagegeneration@002 samples (deprecated model)

Author: glasnt

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(