tests: fix req. explicit field in hashability test

Haidra-Org · Feb 1, 2024 · 4f7148a · 4f7148a
1 parent f1866b5
commit 4f7148a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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(),
     )