fix(preprocessor-video): move sklearn dep to apply

gnes-ai · Jul 12, 2019 · 37155bb · 37155bb
1 parent 2936da7
commit 37155bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gnes/preprocessor/video/shotdetect.py b/gnes/preprocessor/video/shotdetect.py
@@ -16,7 +16,6 @@
 # pylint: disable=low-comment-ratio
 
 import numpy as np
-from sklearn.cluster import KMeans
 from .base import BaseVideoPreprocessor
 from ...proto import gnes_pb2, array2blob
 from ..helper import get_video_frames, compute_descriptor, compare_descriptor
@@ -39,6 +38,7 @@ def __init__(self,
 
     def apply(self, doc: 'gnes_pb2.Document') -> None:
         super().apply(doc)
+        from sklearn.cluster import KMeans
 
         if doc.raw_bytes:
             # stream_data = io.BytesIO(doc.raw_bytes)