From 72f4b9a590a6e432041ca8acb73825c242959d91 Mon Sep 17 00:00:00 2001
From: Jake Poznanski <jakep@allenai.org>
Date: Thu, 30 Jan 2025 15:33:04 -0800
Subject: [PATCH] Project setup

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

diff --git a/pyproject.toml b/pyproject.toml
index a8143e6..325fcaa 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -69,6 +69,8 @@ dev = [
     "sphinx-autodoc-typehints==1.23.3",
     "packaging",
     "necessary",
+    "peft",
+    "datasets"
 ]
 
 
diff --git a/tests/test_sglang.py b/tests/test_sglang.py
index c227c11..399166e 100644
--- a/tests/test_sglang.py
+++ b/tests/test_sglang.py
@@ -31,8 +31,6 @@ 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/"
 )