From 65adfb3f8437ca2ed3ca36367ddda1fa83dffad6 Mon Sep 17 00:00:00 2001 From: tazlin Date: Sat, 17 Feb 2024 12:28:43 -0500 Subject: [PATCH] fix: skipped logic --- horde_sdk/ai_horde_api/apimodels/generate/_pop.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/horde_sdk/ai_horde_api/apimodels/generate/_pop.py b/horde_sdk/ai_horde_api/apimodels/generate/_pop.py index 920bd32..fc25c27 100644 --- a/horde_sdk/ai_horde_api/apimodels/generate/_pop.py +++ b/horde_sdk/ai_horde_api/apimodels/generate/_pop.py @@ -132,7 +132,11 @@ def validate_id(cls, v: str | JobID) -> JobID | str: @model_validator(mode="after") def ids_present(self) -> ImageGenerateJobPopResponse: """Ensure that either id_ or ids is present.""" - if self.skipped.is_empty() and self.model is None: + if self.model is None: + if self.skipped.is_empty(): + logger.debug("No model or skipped data found in response.") + else: + logger.debug("No model found in response.") return self if self.id_ is None and len(self.ids) == 0: