From 38ffe4279d87ab3194132abf253861744252a716 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stanislav=20Luke=C5=A1?= <github@exyi.cz>
Date: Fri, 15 Apr 2022 17:06:00 +0000
Subject: [PATCH] crf_decode: fix broken type annotation

---
 tensorflow_addons/text/crf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow_addons/text/crf.py b/tensorflow_addons/text/crf.py
index 3820b08a94..b0763c9545 100644
--- a/tensorflow_addons/text/crf.py
+++ b/tensorflow_addons/text/crf.py
@@ -524,7 +524,7 @@ def _scan_fn(state, inputs):
 
 def crf_decode(
     potentials: TensorLike, transition_params: TensorLike, sequence_length: TensorLike
-) -> tf.Tensor:
+) -> Tuple[tf.Tensor, tf.Tensor]:
     """Decode the highest scoring sequence of tags.
 
     Args: