Skip imagegen tests in CI

Significant-Gravitas · Apr 19, 2023 · 5b86682 · 5b86682
1 parent 7086961
commit 5b86682
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_image_gen.py b/tests/test_image_gen.py
@@ -1,7 +1,7 @@
 import unittest
 import hashlib
 from PIL import Image
-
+import os
 from autogpt.config import Config
 from autogpt.commands.image_gen import generate_image, generate_image_with_sd_webui
 from autogpt.workspace import path_in_workspace
@@ -11,6 +11,7 @@ def lst(txt):
     return txt.split(":")[1].strip()
 
 
+@unittest.skipIf(os.getenv("CI"), "Skipping image generation tests")
 class TestImageGen(unittest.TestCase):
     def setUp(self):
         self.config = Config()