refine

siliconflow · Aug 8, 2024 · 0805f34 · 0805f34
1 parent 300c51e
commit 0805f34
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -15,6 +15,11 @@
 
 
 def test_comfyui():
+    # Wait for the page to load
+    print(f"wainting for comfyui to start")
+    time.sleep(10)
+    print(f"End wainting for comfyui to start")
+
     chrome_options = Options()
     chrome_options.add_argument("--headless")
     chrome_options.add_argument("--no-sandbox")
@@ -23,9 +28,6 @@ def test_comfyui():
     driver = webdriver.Chrome(options=chrome_options)
     driver.get(f"http://{COMFY_HOST}:{COMFY_PORT}")
 
-    # Wait for the page to load
-    time.sleep(5)
-
     # Add a cookie
     driver.add_cookie({"name": "api_key", "value": BIZYAIR_KEY, "path": "/"})