From 721d1efe03882a3ccc2e0f3c75773ecee28ed559 Mon Sep 17 00:00:00 2001 From: Oliver Fei Date: Thu, 24 Sep 2026 16:01:49 -0400 Subject: [PATCH] fix(guardrails): address TrendAI lint budget --- .../guardrails/guardrail_hooks/trendai/__init__.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/trendai/__init__.py b/litellm/proxy/guardrails/guardrail_hooks/trendai/__init__.py index a506dab07bb..518089f0d02 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/trendai/__init__.py +++ b/litellm/proxy/guardrails/guardrail_hooks/trendai/__init__.py @@ -16,7 +16,7 @@ def _normalize_event_hook( if isinstance(mode, str): return GuardrailEventHooks(mode) if isinstance(mode, list): - return [GuardrailEventHooks(item) for item in mode] + return [GuardrailEventHooks(item) for item in mode] # mutable-ok: guardrail event hooks require a list return mode @@ -41,7 +41,11 @@ def initialize_guardrail(litellm_params: "LitellmParams", guardrail: "Guardrail" return callback -guardrail_initializer_registry: Final = {GUARDRAIL_NAME: initialize_guardrail} -guardrail_class_registry: Final = {GUARDRAIL_NAME: TrendAIGuardrail} +guardrail_initializer_registry: Final = { # mutable-ok: registry discovery requires a dict + GUARDRAIL_NAME: initialize_guardrail +} +guardrail_class_registry: Final = { # mutable-ok: registry discovery requires a dict + GUARDRAIL_NAME: TrendAIGuardrail +} __all__ = ("TrendAIGuardrail",)