Files
sci-gui-agent-benchmark/desktop_env/evaluators/metrics/pdf.py
2024-01-28 21:19:18 +08:00

14 lines
362 B
Python

import operator
from typing import Any
from typing import Dict
from pypdf import PdfReader
def check_pdf_pages(pdf_file: str, rules: Dict[str, Any]) -> float:
if pdf_file is None:
return 0.0
reader = PdfReader(pdf_file)
nb_pages: int = len(reader.pages)
return float(getattr(operator, rules["relation"])(nb_pages, rules["ref_value"]))