Update image.py

vandeplaslab · Feb 2, 2024 · 90b4fe1 · 90b4fe1
1 parent b2379b9
commit 90b4fe1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/koyo/image.py b/src/koyo/image.py
@@ -2,7 +2,6 @@
 import typing as ty
 
 import numpy as np
-from sklearn.metrics.pairwise import cosine_similarity
 
 
 def clip_hotspots(img: np.ndarray, quantile: float = 0.99) -> np.ndarray:
@@ -148,6 +147,7 @@ def colocalization(img_a: np.ndarray, img_b: np.ndarray) -> float:
         similarity score between two images
     """
     from scipy.ndimage import median_filter
+    from sklearn.metrics.pairwise import cosine_similarity
 
     img_a = np.nan_to_num(img_a)
     img_b = np.nan_to_num(img_b)