未验证 提交 a6ba016e 编写于 作者: J Jiangxinz 提交者: GitHub

fix unexpected keyword arg (#33569)

上级 0011450c
...@@ -18,7 +18,8 @@ import unittest ...@@ -18,7 +18,8 @@ import unittest
import numpy as np import numpy as np
from op_test import OpTest from op_test import OpTest
from paddle import fluid from paddle import fluid
from paddle.fluid.layers import lstm, fill_constant from paddle.fluid.layers import lstm as LSTM
from paddle.fluid.layers import fill_constant
from paddle.fluid.framework import program_guard, Program from paddle.fluid.framework import program_guard, Program
SIGMOID_THRESHOLD_MIN = -40.0 SIGMOID_THRESHOLD_MIN = -40.0
...@@ -156,21 +157,21 @@ class LstmUnitTestError(unittest.TestCase): ...@@ -156,21 +157,21 @@ class LstmUnitTestError(unittest.TestCase):
(num_layers, batch_size, hidden_size)).astype('float64') (num_layers, batch_size, hidden_size)).astype('float64')
def test_input_Variable(): def test_input_Variable():
lstm(np_input, pre_hidden, pre_cell, \ LSTM(np_input, pre_hidden, pre_cell, \
seq_len, hidden_size, num_layers, \ seq_len, hidden_size, num_layers, \
dropout_prob=dropout_prob) dropout_prob=dropout_prob)
self.assertRaises(TypeError, test_input_Variable) self.assertRaises(TypeError, test_input_Variable)
def test_pre_hidden_Variable(): def test_pre_hidden_Variable():
lstm(np_input, np_pre_hidden, pre_cell, \ LSTM(np_input, np_pre_hidden, pre_cell, \
seq_len, hidden_size, num_layers, \ seq_len, hidden_size, num_layers, \
dropout_prob=dropout_prob) dropout_prob=dropout_prob)
self.assertRaises(TypeError, test_pre_hidden_Variable) self.assertRaises(TypeError, test_pre_hidden_Variable)
def test_pre_cell_Variable(): def test_pre_cell_Variable():
lstm(np_input, pre_hidden, np_pre_cell, \ LSTM(np_input, pre_hidden, np_pre_cell, \
seq_len, hidden_size, num_layers, \ seq_len, hidden_size, num_layers, \
dropout_prob=dropout_prob) dropout_prob=dropout_prob)
...@@ -181,7 +182,7 @@ class LstmUnitTestError(unittest.TestCase): ...@@ -181,7 +182,7 @@ class LstmUnitTestError(unittest.TestCase):
name='error_input', name='error_input',
shape=[None, hidden_size * 3], shape=[None, hidden_size * 3],
dtype='int32') dtype='int32')
lstm(error_input, pre_hidden, pre_cell, \ LSTM(error_input, pre_hidden, pre_cell, \
seq_len, hidden_size, num_layers, \ seq_len, hidden_size, num_layers, \
dropout_prob=dropout_prob) dropout_prob=dropout_prob)
...@@ -192,7 +193,7 @@ class LstmUnitTestError(unittest.TestCase): ...@@ -192,7 +193,7 @@ class LstmUnitTestError(unittest.TestCase):
name='error_pre_hidden', name='error_pre_hidden',
shape=[None, hidden_size], shape=[None, hidden_size],
dtype='int32') dtype='int32')
lstm(input, error_pre_hidden, pre_cell, \ LSTM(input, error_pre_hidden, pre_cell, \
seq_len, hidden_size, num_layers, \ seq_len, hidden_size, num_layers, \
dropout_prob=dropout_prob) dropout_prob=dropout_prob)
...@@ -203,7 +204,7 @@ class LstmUnitTestError(unittest.TestCase): ...@@ -203,7 +204,7 @@ class LstmUnitTestError(unittest.TestCase):
name='error_pre_cell', name='error_pre_cell',
shape=[None, hidden_size], shape=[None, hidden_size],
dtype='int32') dtype='int32')
lstm(input, pre_hidden, error_pre_cell, \ LSTM(input, pre_hidden, error_pre_cell, \
seq_len, hidden_size, num_layers, \ seq_len, hidden_size, num_layers, \
dropout_prob=dropout_prob) dropout_prob=dropout_prob)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册