From 2a399e29ddc714f1173a6d9a3076d83b6fe329f8 Mon Sep 17 00:00:00 2001 From: db0 Date: Tue, 9 Jan 2024 19:36:33 +0100 Subject: [PATCH] fix: handle when image_result is None --- horde_worker_regen/process_management/process_manager.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py index a7869076..1e940dae 100644 --- a/horde_worker_regen/process_management/process_manager.py +++ b/horde_worker_regen/process_management/process_manager.py @@ -1666,7 +1666,9 @@ async def submit_single_generation(self, new_submit: PendingSubmitJob) -> Pendin logger.error(f"Failed to upload image to R2: {response}") new_submit.retry() return new_submit - + metadata = [] + if new_submit.image_result is not None: + metadata = new_submit.image_result.generation_faults submit_job_request_type = new_submit.completed_job_info.sdk_api_job_info.get_follow_up_default_request_type() submit_job_request = submit_job_request_type( apikey=self.bridge_data.api_key, @@ -1675,7 +1677,7 @@ async def submit_single_generation(self, new_submit: PendingSubmitJob) -> Pendin generation="R2", # TODO # FIXME state=new_submit.completed_job_info.state, censored=bool(new_submit.completed_job_info.censored), # TODO: is this cast problematic? - gen_metadata=new_submit.image_result.generation_faults, + gen_metadata=metadata, ) job_submit_response = await self.horde_client_session.submit_request( submit_job_request,