Skip to content
This repository has been archived by the owner on Feb 22, 2020. It is now read-only.

Commit

Permalink
fix(preprocessor-video): move sklearn dep to apply
Browse files Browse the repository at this point in the history
  • Loading branch information
Han Xiao committed Jul 12, 2019
1 parent 2936da7 commit 37155bb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion gnes/preprocessor/video/shotdetect.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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)
Expand Down

0 comments on commit 37155bb

Please sign in to comment.