From 6991838a3176c6584c5908ec4f2701666d4a5e57 Mon Sep 17 00:00:00 2001 From: Thomas Werkmeister Date: Tue, 26 Jul 2022 15:06:26 +0200 Subject: [PATCH] Fixed unintended change in dynamic shape calculation --- rasa/utils/tensorflow/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rasa/utils/tensorflow/models.py b/rasa/utils/tensorflow/models.py index b5a42dc1585f..461ad3b9914b 100644 --- a/rasa/utils/tensorflow/models.py +++ b/rasa/utils/tensorflow/models.py @@ -234,7 +234,7 @@ def _dynamic_signature( element_spec = [] for tensor in batch_in: if len(tensor.shape) > 1: - shape: List[Union[None, int]] = [None] * (len(tensor.shape) - 2) + shape: List[Union[None, int]] = [None] * (len(tensor.shape) - 1) shape += [tensor.shape[-1]] else: shape = [None]