From a1f5abe8e1855d1d5a18870a5d3129f3c2040f15 Mon Sep 17 00:00:00 2001
From: weipingpku <630404796@qq.com>
Date: Thu, 4 May 2017 00:48:21 +0800
Subject: [PATCH] Fix typo in LINE#500 (#9626)

cell_input_fn -> probability_fn
---
 tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py b/tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py
index 9fc548aabe3..a0f1775257d 100644
--- a/tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py
+++ b/tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py
@@ -497,7 +497,7 @@ class AttentionWrapper(core_rnn_cell.RNNCell):
     if probability_fn is None:
       probability_fn = nn_ops.softmax
     else:
-      if not callable(cell_input_fn):
+      if not callable(probability_fn):
         raise TypeError(
             "probability_fn must be callable, saw type: %s"
             % type(probability_fn).__name__)