From f30e6007c04b6ee5612c5957d698ad184f47a567 Mon Sep 17 00:00:00 2001 From: tazlin Date: Mon, 24 Jun 2024 10:17:06 -0400 Subject: [PATCH] tests: log `skipped` separate w/ fake worker --- tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py | 5 +++-- 1 file changed, 3 insertions(+), 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 index 8394161..f43de89 100644 --- 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: