From 11302feb8cbe0b3fe3463aa6755f0b148f80c4d2 Mon Sep 17 00:00:00 2001
From: Jake Poznanski <jakep@allenai.org>
Date: Wed, 13 Aug 2025 20:28:31 +0000
Subject: [PATCH] Move open cv2 import only into experimental data loader class

---
 olmocr/train/dataloader.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/olmocr/train/dataloader.py b/olmocr/train/dataloader.py
index c7958f3..d7637e1 100644
--- a/olmocr/train/dataloader.py
+++ b/olmocr/train/dataloader.py
@@ -21,7 +21,6 @@ from typing import (
     get_origin,
 )
 
-import cv2
 import numpy as np
 import yaml
 from PIL import Image
@@ -533,6 +532,9 @@ class AugraphyBasicAugmentations(PipelineStep):
         if "image" not in sample:
             return sample
 
+        # Import opencv only here
+        import cv2
+
         image = sample["image"]
 
         # Skip all augmentations based on overall probability