From 7af8fb581b6617110283fc6a94843c9e5ae6e8a0 Mon Sep 17 00:00:00 2001 From: Fabio Manganiello Date: Mon, 23 Mar 2020 23:54:18 +0100 Subject: [PATCH] A more robust logic to check whether the first dimension of the input tensor is None --- platypush/plugins/tensorflow/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/platypush/plugins/tensorflow/__init__.py b/platypush/plugins/tensorflow/__init__.py index 55d39df7a..83a68f3f3 100644 --- a/platypush/plugins/tensorflow/__init__.py +++ b/platypush/plugins/tensorflow/__init__.py @@ -1010,7 +1010,7 @@ class TensorflowPlugin(Plugin): inputs = self._get_data(inputs, model) if isinstance(inputs, np.ndarray) and \ len(model.inputs[0].shape) == len(inputs.shape) + 1 and \ - model.inputs[0].shape[0] is None: + (model.inputs[0].shape[0] is None or model.inputs[0].shape[0].value is None): inputs = np.asarray([inputs]) ret = model.predict(