deprecate tests for older stable diffusion models.

Author: jfacevedo-google

src/maxdiffusion/tests/generate_smoke_test.py

@@ -22,7 +22,7 @@ def setUp(self):
     super().setUp()
     Generate.dummy_data = {}
 
-  @pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Don't run smoke tests on Github Actions")
+  @pytest.mark.skip("This test is deprecated and will be removed in a future version.")
   def test_sd14_config(self):
     img_url = os.path.join(THIS_DIR, "images", "test_gen_sd14.png")
     base_image = np.array(Image.open(img_url)).astype(np.uint8)
@@ -42,7 +42,7 @@ def test_sd14_config(self):
     assert base_image.shape == test_image.shape
     assert ssim_compare >= 0.70
 
-  @pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Don't run smoke tests on Github Actions")
+  @pytest.mark.skip("This test is deprecated and will be removed in a future version.")
   def test_sd_2_base_from_gcs(self):
     img_url = os.path.join(THIS_DIR, "images", "test_2_base.png")
     base_image = np.array(Image.open(img_url)).astype(np.uint8)
@@ -64,7 +64,7 @@ def test_sd_2_base_from_gcs(self):
     assert base_image.shape == test_image.shape
     assert ssim_compare >= 0.70
 
-  @pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Don't run smoke tests on Github Actions")
+  @pytest.mark.skip("This test is deprecated and will be removed in a future version.")
   def test_controlnet(self):
     img_url = os.path.join(THIS_DIR, "images", "cnet_test.png")
     base_image = np.array(Image.open(img_url)).astype(np.uint8)

src/maxdiffusion/tests/input_pipeline_interface_test.py

@@ -20,6 +20,7 @@
 import shutil
 import subprocess
 import unittest
+import pytest
 from absl.testing import absltest
 import numpy as np
 import tensorflow as tf
@@ -431,6 +432,7 @@ def test_make_pokemon_iterator_sdxl_cache(self):
         config.resolution // vae_scale_factor,
     )
 
+  @pytest.mark.skip("This test is deprecated and will be removed in a future version. Reason: stable diffusion 2 base is no longer in HuggingFace")
   def test_make_laion_grain_iterator(self):
     try:
       subprocess.check_output(
@@ -486,7 +488,8 @@ def test_make_laion_grain_iterator(self):
         config.resolution // vae_scale_factor,
         8,
     )
-
+    
+  @pytest.mark.skip("This test is deprecated and will be removed in a future version. Reason: stable diffusion 2 base is no longer in HuggingFace")
   def test_make_laion_tfrecord_iterator(self):
     pyconfig.initialize(
         [

src/maxdiffusion/tests/train_smoke_test.py

@@ -96,7 +96,7 @@ def test_sdxl_config(self):
 
     delete_blobs(os.path.join(output_dir, run_name))
 
-  @pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Don't run smoke tests on Github Actions")
+  @pytest.mark.skip("This test is deprecated and will be removed in a future version.")
   def test_dreambooth_orbax(self):
     num_class_images = 100
     output_dir = "gs://maxdiffusion-github-runner-test-assets"
@@ -149,7 +149,7 @@ def test_dreambooth_orbax(self):
     cleanup(class_class_local_dir)
     delete_blobs(os.path.join(output_dir, run_name))
 
-  @pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Don't run smoke tests on Github Actions")
+  @pytest.mark.skip("This test is deprecated and will be removed in a future version.")
   def test_sd15_orbax(self):
     output_dir = "gs://maxdiffusion-github-runner-test-assets"
     run_name = "sd15_orbax_smoke_test"