HotFix: opencv import placement

opencv should only be imported when used due to its complications in installation
CBroz1 · Feb 3, 2024 · b925dff · b925dff
1 parent e8ae471
commit b925dff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/spyglass/position/v1/position_trodes_position.py b/src/spyglass/position/v1/position_trodes_position.py
@@ -2,7 +2,6 @@
 import os
 from pathlib import Path
 
-import cv2
 import datajoint as dj
 import numpy as np
 from datajoint.utils import to_camel_case
@@ -357,6 +356,8 @@ def make_video(
         arrow_radius=15,
         circle_radius=8,
     ):
+        import cv2
+
         RGB_PINK = (234, 82, 111)
         RGB_YELLOW = (253, 231, 76)
         RGB_WHITE = (255, 255, 255)