fix(test): fix import

gnes-ai · Aug 2, 2019 · 82c5005 · 82c5005
1 parent c1e406a
commit 82c5005
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_onnx_image_encoder.py b/tests/test_onnx_image_encoder.py
@@ -4,7 +4,7 @@
 import zipfile
 
 from gnes.encoder.image.onnx import BaseONNXImageEncoder
-from gnes.preprocessor.base import BaseUnaryPreprocessor
+from gnes.preprocessor.base import BaseUnaryPreprocessor, PipelinePreprocessor
 from gnes.preprocessor.image.sliding_window import VanillaSlidingPreprocessor
 from gnes.proto import gnes_pb2, blob2array