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