def rnn_cell_test(f):
f = keras_test(f)
return pytest.mark.parametrize('cell_class', [
recurrent.SimpleRNNCell,
recurrent.GRUCell,
recurrent.LSTMCell
])(f)
def rnn_cell_test(f):
f = keras_test(f)
return pytest.mark.parametrize('cell_class', [
recurrent.SimpleRNNCell,
recurrent.GRUCell,
recurrent.LSTMCell
])(f)
def rnn_test(f):
"""
All the recurrent layers share the same interface,
so we can run through them with a single function.
"""
f = keras_test(f)
return pytest.mark.parametrize(
'layer_class', [recurrent.SimpleRNN, recurrent.GRU, recurrent.LSTM])(f)
def rnn_test(f):
"""
All the recurrent layers share the same interface,
so we can run through them with a single function.
"""
f = keras_test(f)
# Example : return pytest.mark.parametrize("layer_class", [recurrent.JZ1, recurrent.NTM])(f)
return pytest.mark.parametrize('layer_class', [])(f)
def rnn_test(f):
"""
All the recurrent layers share the same interface,
so we can run through them with a single function.
"""
f = keras_test(f)
return pytest.mark.parametrize('layer_class', [
recurrent.SimpleRNN,
recurrent.GRU,
recurrent.LSTM
])(f)
def rnn_test(f):
"""
All the recurrent layers share the same interface,
so we can run through them with a single function.
"""
kf = keras_test(f)
def wrapped(layer_class):
return kf(layer_class)
# functools doesnt propagate arguments info for pytest correctly in 2.7
# and wrapped doesnt work with pytest in 3.4
if sys.version_info >= (3, 0):
f = kf
else:
f = wrapped
return pytest.mark.parametrize(
"layer_class", [recurrent.SimpleRNN, recurrent.GRU, recurrent.LSTM])(f)
def rnn_test(f):
"""
All the recurrent layers share the same interface,
so we can run through them with a single function.
"""
kf = keras_test(f)
def wrapped(layer_class):
return kf(layer_class)
# functools doesnt propagate arguments info for pytest correctly in 2.7
# and wrapped doesnt work with pytest in 3.4
if sys.version_info >= (3, 0):
f = kf
else:
f = wrapped
return pytest.mark.parametrize("layer_class", [
recurrent.SimpleRNN,
recurrent.GRU,
recurrent.LSTM
])(f)