From 471f011ae50424482b0bbce012323bb5c8d8b7e0 Mon Sep 17 00:00:00 2001 From: Arshdeep Singh Date: Thu, 25 Sep 2025 12:52:12 -0400 Subject: [PATCH] fix: add return type annotation for full mypy compliance --- .../proxy/guardrails/guardrail_hooks/test_presidio_union_fix.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_presidio_union_fix.py b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_presidio_union_fix.py index ff31249665..841288948d 100644 --- a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_presidio_union_fix.py +++ b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_presidio_union_fix.py @@ -15,7 +15,7 @@ class PiiAction(str, Enum): MASK = "MASK" -def test_presidio_union_type_fix(): +def test_presidio_union_type_fix() -> None: """Test that Union[PiiEntityType, str] allows both enum and string entity types""" # This is the core fix - mixed entity types in pii_entities_config