apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): add progress bars to release tests

This commit is contained in:
Sean Sube 2023-07-02 13:18:00 -05:00
parent be396099ef
commit 5d18779e27
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/scripts/test-release.py
View File

@ -5,6 +5,7 @@ from logging import getLogger
from logging.config import dictConfig from logging.config import dictConfig
from os import environ, path from os import environ, path
from time import sleep from time import sleep
from tqdm import tqdm
from typing import List, Optional, Union from typing import List, Optional, Union
import cv2 import cv2
@ -217,6 +218,7 @@ TEST_DATA = [
"upscale-sd-x4-2048-muffin", "upscale-sd-x4-2048-muffin",
"upscale?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&upscaling=upscaling-stable-diffusion-x4&scale=4&outscale=4", "upscale?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&upscaling=upscaling-stable-diffusion-x4&scale=4&outscale=4",
source="txt2img-sd-v1-5-512-muffin-0", source="txt2img-sd-v1-5-512-muffin-0",
max_attempts=VERY_SLOW_TEST,
), ),
TestCase( TestCase(
"outpaint-panorama-even-256", "outpaint-panorama-even-256",
@ -290,6 +292,7 @@ TEST_DATA = [
"&correction=correction-codeformer&faces=true&faceOutscale=1&faceStrength=1.0" "&correction=correction-codeformer&faces=true&faceOutscale=1&faceStrength=1.0"
), ),
source="txt2img-sd-v1-5-512-muffin-0", source="txt2img-sd-v1-5-512-muffin-0",
max_attempts=VERY_SLOW_TEST,
), ),
TestCase( TestCase(
"upscale-sd-x4-gfpgan-2048-muffin", "upscale-sd-x4-gfpgan-2048-muffin",
@ -298,6 +301,7 @@ TEST_DATA = [
"&correction=correction-gfpgan&faces=true&faceOutscale=1&faceStrength=1.0" "&correction=correction-gfpgan&faces=true&faceOutscale=1&faceStrength=1.0"
), ),
source="txt2img-sd-v1-5-512-muffin-0", source="txt2img-sd-v1-5-512-muffin-0",
max_attempts=VERY_SLOW_TEST,
), ),
TestCase( TestCase(
"txt2img-panorama-1024x768-muffin", "txt2img-panorama-1024x768-muffin",
@ -446,17 +450,17 @@ def run_test(
if keys is None: if keys is None:
raise ValueError("could not generate image") raise ValueError("could not generate image")
attempts = 0 ready = False
while attempts < test.max_attempts: for attempt in tqdm(range(test.max_attempts)):
if check_ready(host, keys[0]): if check_ready(host, keys[0]):
logger.debug("image is ready: %s", keys) logger.debug("image is ready: %s", keys)
ready = True
break break
else: else:
logger.debug("waiting for image to be ready") logger.debug("waiting for image to be ready")
attempts += 1
sleep(6) sleep(6)
if attempts == test.max_attempts: if not ready:
raise ValueError("image was not ready in time") raise ValueError("image was not ready in time")
results = download_images(host, keys) results = download_images(host, keys)