From 9db85d02d6a6be0b2c302e322702757a8bdc496e Mon Sep 17 00:00:00 2001
From: YangBin1729 <50950218+YangBin1729@users.noreply.github.com>
Date: Wed, 20 May 2020 18:24:49 +0800
Subject: [PATCH] annotation error

---
 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 4ddcc1dcac..c1bde04e0d 100644
--- a/tensorflow_addons/text/crf.py
+++ b/tensorflow_addons/text/crf.py
@@ -309,7 +309,7 @@ def crf_forward(
     See http://www.cs.columbia.edu/~mcollins/fb.pdf for reference.
 
     Args:
-      inputs: A [batch_size, num_tags] matrix of unary potentials.
+      inputs: A [batch_size, max_seq_len, num_tags] matrix of unary potentials.
       state: A [batch_size, num_tags] matrix containing the previous alpha
          values.
       transition_params: A [num_tags, num_tags] matrix of binary potentials.