test: mark flaky tests

This commit is contained in:
Krrish Dholakia 2025-04-23 21:50:16 -07:00
parent a260afb74d
commit 2486a106f4
2 changed files with 7 additions and 0 deletions

View file

@ -311,6 +311,9 @@ def exception_type( # type: ignore # noqa: PLR0915
elif (
"invalid_request_error" in error_str
and "content_policy_violation" in error_str
) or (
"Invalid prompt" in error_str
and "violating our usage policy" in error_str
):
exception_mapping_worked = True
raise ContentPolicyViolationError(