diff --git a/benchmark.py b/benchmark.py index 445e2ad..9f70f76 100644 --- a/benchmark.py +++ b/benchmark.py @@ -80,7 +80,7 @@ def main(): row = dataset[i] pdf = row["pdf"] tool_pages = {} - with tempfile.NamedTemporaryFile(suffix=".pdf", mode='w') as f: + with tempfile.NamedTemporaryFile(suffix=".pdf", mode='wb') as f: f.write(pdf) f.seek(0) pdf_path = f.name