diff --git a/src/frontend_visualqa/schemas.py b/src/frontend_visualqa/schemas.py index 4e08e41..78e393a 100644 --- a/src/frontend_visualqa/schemas.py +++ b/src/frontend_visualqa/schemas.py @@ -213,14 +213,7 @@ def validate_claims(cls, value: list[str]) -> list[str]: def normalize_claim_navigation_hints(cls, value: list[str | None] | None) -> list[str | None] | None: if value is None: return None - normalized: list[str | None] = [] - for hint in value: - if hint is None: - normalized.append(None) - continue - stripped = hint.strip() - normalized.append(stripped or None) - return normalized + return [None if hint is None else (hint.strip() or None) for hint in value] @model_validator(mode="after") def validate_claim_navigation_hint_alignment(self) -> "VerifyVisualClaimsInput":