diff --git a/tests/e2e/budgets/test_model_max_budget_e2e.py b/tests/e2e/budgets/test_model_max_budget_e2e.py index 34616624c12..c11e50f679c 100644 --- a/tests/e2e/budgets/test_model_max_budget_e2e.py +++ b/tests/e2e/budgets/test_model_max_budget_e2e.py @@ -24,7 +24,7 @@ FREE_MODEL = "gemini-2.5-flash" def _call(client: BudgetClient, key: str, model: str): result = client.chat(key, model, f"hi {unique_marker()}", max_tokens=16) if not result.ok and not is_budget_block(result): - require_successful_call(result) # non-budget error -> skip + require_successful_call(result) return result diff --git a/tests/e2e/budgets/test_tag_budget_e2e.py b/tests/e2e/budgets/test_tag_budget_e2e.py index 6f71b1b204b..492aa89b528 100644 --- a/tests/e2e/budgets/test_tag_budget_e2e.py +++ b/tests/e2e/budgets/test_tag_budget_e2e.py @@ -29,7 +29,7 @@ def _tagged_call(client: BudgetClient, key: str, tag: str): max_tokens=16, ) if not result.ok and not is_budget_block(result): - require_successful_call(result) # non-budget error -> skip + require_successful_call(result) return result