yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-12-27 15:14:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

added Api_key argument to pipeline pytests

Browse Source
This commit is contained in:
aman-17 2025-09-26 14:25:25 -07:00
parent 359abef654
commit f3c4073395
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_pipeline.py
View File

@ -209,7 +209,7 @@ class TestRotationCorrection:
# Counter to track number of API calls
call_count = 0
async def mock_apost(url, json_data):
async def mock_apost(url, json_data, api_key=None):
nonlocal call_count
call_count += 1
@ -311,7 +311,7 @@ This is the corrected text from the document."""
# Counter to track number of API calls
call_count = 0
async def mock_apost(url, json_data):
async def mock_apost(url, json_data, api_key=None):
nonlocal call_count
call_count += 1
@ -420,7 +420,7 @@ Document is now correctly oriented after 180 degree rotation."""
# Counter to track number of API calls
call_count = 0
async def mock_apost(url, json_data):
async def mock_apost(url, json_data, api_key=None):
nonlocal call_count
call_count += 1