From aa239eb34c94a42292f197cf9880cda0345d02ff Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Mon, 13 Oct 2025 21:15:19 +0000 Subject: [PATCH] Lints --- olmocr/train/grpo_train.py | 2 +- tests/test_olmocrmix.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/olmocr/train/grpo_train.py b/olmocr/train/grpo_train.py index 4d25c2f..0292ab8 100644 --- a/olmocr/train/grpo_train.py +++ b/olmocr/train/grpo_train.py @@ -18,7 +18,7 @@ import torch import torch.distributed as dist import wandb from PIL import Image -from rapidfuzz import fuzz, distance +from rapidfuzz import distance, fuzz from torch.utils.data import Dataset from transformers import ( AutoProcessor, diff --git a/tests/test_olmocrmix.py b/tests/test_olmocrmix.py index a31d4e6..b837c1f 100644 --- a/tests/test_olmocrmix.py +++ b/tests/test_olmocrmix.py @@ -84,8 +84,8 @@ def test_repackage_and_prepare_olmocrmix(): if relative_path.suffix == ".jsonl": # For JSONL files, compare as sets of lines (order doesn't matter) # Filter out empty lines - sample_lines = set(line for line in sample_file.read_text().strip().split('\n') if line.strip()) - unpacked_lines = set(line for line in unpacked_file.read_text().strip().split('\n') if line.strip()) + sample_lines = set(line for line in sample_file.read_text().strip().split("\n") if line.strip()) + unpacked_lines = set(line for line in unpacked_file.read_text().strip().split("\n") if line.strip()) assert sample_lines == unpacked_lines, f"JSONL file contents differ for {relative_path}" else: # For other files, compare as bytes