yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-11-02 11:04:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Type checks better

Browse Source
This commit is contained in:
Jake Poznanski 2025-08-21 18:51:05 +00:00
parent 33d889c748
commit 6e3e2d8abc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
olmocr/train/grpo_train.py
View File

@ -219,7 +219,7 @@ def olmocr_bench_reward(prompts, completions: list[str] | list[list[dict]], comp
logger.info(f"Completion {i}: PDF: {comp_pdf_path}, JSONL: {comp_jsonl_file}, Test IDs: {comp_test_ids}")
if completion is None or not isinstance(completion, str) or not isinstance(completion, list):
if completion is None or not (isinstance(completion, str) or isinstance(completion, list)):
logger.warning(f"Invalid completion at index {i}: {type(completion)}")
logger.warning(f"completion: {completion}")
rewards.append(None)