From 4fd2ff2aceab00e97e82c7d4a2e835fe9a16d23c Mon Sep 17 00:00:00 2001 From: Anish Date: Sat, 27 Jun 2020 09:27:50 +0530 Subject: [PATCH] fix(executor): fix init Co-authored-by: Nan Wang --- jina/executors/encoders/image/customtfkeras.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jina/executors/encoders/image/customtfkeras.py b/jina/executors/encoders/image/customtfkeras.py index a3ac804e5277a..f730f00befdce 100644 --- a/jina/executors/encoders/image/customtfkeras.py +++ b/jina/executors/encoders/image/customtfkeras.py @@ -16,7 +16,7 @@ class CustomImageKerasEncoder(BaseCVTFEncoder): https://www.tensorflow.org/api_docs/python/tf/keras/applications """ - def __init__(self, model_path: str, layer_name: str, *args, **kwargs): + def __init__(self, model_path: str, layer_name: str, channel_axis: int = -1, *args, **kwargs): """ :param model_path: the path where the model is stored.