Update tests

jpuigcerver · Oct 4, 2023 · eb04c71 · eb04c71
1 parent 812a3a7
commit eb04c71
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)],
         }