refactor(executor): fix init

jina-ai · Jun 26, 2020 · 02e02dc · 02e02dc
1 parent 58e9920
commit 02e02dc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/jina/executors/encoders/image/customtfkeras.py b/jina/executors/encoders/image/customtfkeras.py
@@ -15,6 +15,8 @@ class CustomImageKerasEncoder(BaseCVTFEncoder):
     Internally, :class:`CustomImageKerasEncoder` wraps any custom tf.keras model not part of models from `tensorflow.keras.applications`.
     https://www.tensorflow.org/api_docs/python/tf/keras/applications
     """
+
+     def __init__(self, model_path: str, layer_name: str, *args, **kwargs):
 
         """
         :param model_path: the path where the model is stored.