diff --git a/verifiers/rubrics/math_rubric.py b/verifiers/rubrics/math_rubric.py index 7fb2444ba..042ebff61 100644 --- a/verifiers/rubrics/math_rubric.py +++ b/verifiers/rubrics/math_rubric.py @@ -12,6 +12,21 @@ from verifiers.utils.data_utils import extract_boxed_answer +class _TimeoutWarningFilter(logging.Filter): + """Filter to suppress math_verify timeout disabled warnings.""" + + def filter(self, record: logging.LogRecord) -> bool: + # Suppress the specific warning about timeout being disabled + if "Timeout is disabled" in record.getMessage(): + return False + return True + + +# Apply the filter once at module load time to ensure it's set before any MathRubric usage +_grader_logger = logging.getLogger("math_verify.grader") +_grader_logger.addFilter(_TimeoutWarningFilter()) + + class MathRubric(Rubric): def __init__( self,