tests: log skipped separate w/ fake worker

Haidra-Org · Jul 7, 2024 · f30e600 · f30e600
1 parent 855f2e6
commit f30e600
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py b/tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py
@@ -68,7 +68,7 @@ async def fake_worker_checkin(
         )
 
         assert isinstance(job_pop_response, ImageGenerateJobPopResponse)
-        logger.info(f"{job_pop_response.log_safe_model_dump()}")
+        logger.info(f"{job_pop_response.log_safe_model_dump({'skipped'})}")
 
         assert not job_pop_response.ids_present
         assert job_pop_response.skipped is not None
@@ -103,7 +103,8 @@ async def fake_worker(
             )
 
             assert isinstance(job_pop_response, ImageGenerateJobPopResponse)
-            logger.info(f"{job_pop_response.log_safe_model_dump()}")
+            logger.info(f"{job_pop_response.log_safe_model_dump({'skipped'})}")
+            logger.info(f"Checked in as fake worker ({effective_resolution}): {job_pop_response.skipped}")
 
             if not job_pop_response.ids_present:
                 if current_try >= max_tries: