From 10094ffc19c1281669cec3263bd1d4e2e7e51bbe Mon Sep 17 00:00:00 2001
From: Jake Poznanski <jakep@allenai.org>
Date: Thu, 30 Jan 2025 14:32:08 -0800
Subject: [PATCH] Even newer mypy crashes still

---
 pyproject.toml       | 2 +-
 tests/test_sglang.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 9949e25..a8143e6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -51,7 +51,7 @@ Changelog = "https://github.com/allenai/olmocr/blob/main/CHANGELOG.md"
 [project.optional-dependencies]
 dev = [
     "ruff",
-    "mypy>=1.0,<1.5",
+    "mypy",
     "black",
     "isort",
     "pytest",
diff --git a/tests/test_sglang.py b/tests/test_sglang.py
index b3a67fa..c227c11 100644
--- a/tests/test_sglang.py
+++ b/tests/test_sglang.py
@@ -24,7 +24,6 @@ from transformers import AutoProcessor, AutoTokenizer, Qwen2VLForConditionalGene
 from olmocr.pipeline import (
     SGLANG_SERVER_PORT,
     build_page_query,
-    download_directory,
     get_anchor_text,
     render_pdf_to_base64png,
     sglang_server_ready,
@@ -32,6 +31,8 @@ from olmocr.pipeline import (
 )
 from olmocr.prompts import PageResponse
 
+
+
 MODEL_FINETUNED_PATH = (
     "s3://ai2-oe-data/jakep/experiments/qwen2vl-pdf/v1/models/jakep/Qwen_Qwen2-VL-7B-Instruct-e4ecf8-01JAH8GMWHTJ376S2N7ETXRXH4/checkpoint-9500/bf16/"
 )