From f58694e217feb04c6835b1fc03215bce715f5a59 Mon Sep 17 00:00:00 2001 From: Panguard AI Date: Mon, 18 May 2026 17:36:00 +0800 Subject: [PATCH] chore: apply black formatting --- litellm/proxy/guardrails/guardrail_hooks/atr/atr.py | 4 +++- .../proxy/guardrails/guardrail_hooks/test_atr.py | 8 ++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/atr/atr.py b/litellm/proxy/guardrails/guardrail_hooks/atr/atr.py index eba1814028b..620dcc4cc49 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/atr/atr.py +++ b/litellm/proxy/guardrails/guardrail_hooks/atr/atr.py @@ -304,7 +304,9 @@ class ATRGuardrail(CustomGuardrail): # Treat None or unrecognised severity conservatively (rank 0 = critical) raw_severity = getattr(m, "severity", None) - severity_str = (raw_severity or "").lower() if raw_severity is not None else "" + severity_str = ( + (raw_severity or "").lower() if raw_severity is not None else "" + ) rank = _SEVERITY_RANK.get(severity_str, 0) if rank <= threshold_rank: result.append(m) diff --git a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_atr.py b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_atr.py index 3d52fbfa06f..1e06724768a 100644 --- a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_atr.py +++ b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_atr.py @@ -428,7 +428,9 @@ def test_scan_none_severity_treated_conservatively(fake_pyatr, tmp_path): rules_dir = tmp_path / "rules" rules_dir.mkdir() - none_severity_match = MagicMock(rule_id="ATR-600", title="Unknown sev", severity=None) + none_severity_match = MagicMock( + rule_id="ATR-600", title="Unknown sev", severity=None + ) engine.evaluate.return_value = [none_severity_match] guard = ATRGuardrail( @@ -450,7 +452,9 @@ def test_scan_unknown_severity_treated_conservatively(fake_pyatr, tmp_path): rules_dir = tmp_path / "rules" rules_dir.mkdir() - unknown_match = MagicMock(rule_id="ATR-601", title="Future sev", severity="informational") + unknown_match = MagicMock( + rule_id="ATR-601", title="Future sev", severity="informational" + ) engine.evaluate.return_value = [unknown_match] guard = ATRGuardrail(