From eb04c71e511111115a7aa6682f09721bfa37103d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sol=C3=A8ne=20Tarride?= Date: Wed, 4 Oct 2023 16:41:42 +0200 Subject: [PATCH] Update tests --- tests/callbacks/decode_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/callbacks/decode_test.py b/tests/callbacks/decode_test.py index c2750970..4990c0b6 100644 --- a/tests/callbacks/decode_test.py +++ b/tests/callbacks/decode_test.py @@ -15,6 +15,7 @@ def __call__(self, batch_y_hat): "prob-htr-char": [ [0.9, 0.9, 0.9, 0.9, 0.9, 0.9] for _ in range(batch_size) ], + "prob-htr": [0.9 for _ in range(batch_size)], }