diff --git a/tests/ai_horde_api/test_ai_horde_api_models.py b/tests/ai_horde_api/test_ai_horde_api_models.py index 55d0e7d..a0d71af 100644 --- a/tests/ai_horde_api/test_ai_horde_api_models.py +++ b/tests/ai_horde_api/test_ai_horde_api_models.py @@ -423,6 +423,7 @@ def test_ImageGenerateJobPopResponse_hashability() -> None: post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus], prompt="A cat in a hat", ), + model="Deliberate", source_image="r2 download link", skipped=ImageGenerateJobPopSkippedStatus(), ) @@ -434,6 +435,7 @@ def test_ImageGenerateJobPopResponse_hashability() -> None: post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus], prompt="A cat in a hat", ), + model="Deliberate", source_image="parsed base64", skipped=ImageGenerateJobPopSkippedStatus(), ) @@ -445,6 +447,7 @@ def test_ImageGenerateJobPopResponse_hashability() -> None: post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus], prompt="A cat in a hat", ), + model="Deliberate", skipped=ImageGenerateJobPopSkippedStatus(), ) @@ -468,6 +471,7 @@ def test_ImageGenerateJobPopResponse_hashability() -> None: post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus], prompt="A cat in a hat", ), + model="Deliberate", skipped=ImageGenerateJobPopSkippedStatus(), ) @@ -478,6 +482,7 @@ def test_ImageGenerateJobPopResponse_hashability() -> None: post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus], prompt="A cat in a hat", ), + model="Deliberate", skipped=ImageGenerateJobPopSkippedStatus(), ) @@ -503,6 +508,7 @@ def test_ImageGenerateJobPopResponse_hashability() -> None: post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus], prompt="A cat in a hat", ), + model="Deliberate", source_image="r2 download link", skipped=ImageGenerateJobPopSkippedStatus(), ) @@ -517,6 +523,7 @@ def test_ImageGenerateJobPopResponse_hashability() -> None: post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus], prompt="A cat in a hat", ), + model="Deliberate", source_image="r2 download link", skipped=ImageGenerateJobPopSkippedStatus(), ) @@ -531,6 +538,7 @@ def test_ImageGenerateJobPopResponse_hashability() -> None: post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus], prompt="A cat in a hat", ), + model="Deliberate", source_image="parsed base64", skipped=ImageGenerateJobPopSkippedStatus(), )