Skip to content

Commit dcb5a1e

Browse files
committed
fix ci.
1 parent 64606ea commit dcb5a1e

File tree

4 files changed

+30
-19
lines changed

4 files changed

+30
-19
lines changed

python/paddle/v2/dataset/wmt16.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -171,8 +171,9 @@ def train(src_dict_size, trg_dict_size, src_lang="en"):
171171
callable: The train reader.
172172
"""
173173

174-
assert (src_lang in ["en", "de"], ("An error language type. Only support: "
175-
"en (for English); de(for Germany)"))
174+
if src_lang not in ["en", "de"]:
175+
raise ValueError("An error language type. Only support: "
176+
"en (for English); de(for Germany).")
176177
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
177178
src_lang)
178179

@@ -218,9 +219,9 @@ def test(src_dict_size, trg_dict_size, src_lang="en"):
218219
callable: The test reader.
219220
"""
220221

221-
assert (src_lang in ["en", "de"],
222-
("An error language type. "
223-
"Only support: en (for English); de(for Germany)"))
222+
if src_lang not in ["en", "de"]:
223+
raise ValueError("An error language type. "
224+
"Only support: en (for English); de(for Germany).")
224225

225226
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
226227
src_lang)
@@ -266,9 +267,9 @@ def validation(src_dict_size, trg_dict_size, src_lang="en"):
266267
Returns:
267268
callable: The validation reader.
268269
"""
269-
assert (src_lang in ["en", "de"],
270-
("An error language type. "
271-
"Only support: en (for English); de(for Germany)"))
270+
if src_lang not in ["en", "de"]:
271+
raise ValueError("An error language type. "
272+
"Only support: en (for English); de(for Germany).")
272273
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
273274
src_lang)
274275

python/paddle/v2/fluid/layers/nn.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2141,7 +2141,7 @@ def sequence_reshape(input, new_dim):
21412141
return out
21422142

21432143

2144-
def transpose(input, perm, name=None):
2144+
def transpose(x, perm, name=None):
21452145
"""
21462146
**transpose Layer**
21472147
@@ -2161,19 +2161,19 @@ def transpose(input, perm, name=None):
21612161
.. code-block:: python
21622162
21632163
x = fluid.layers.data(name='x', shape=[5, 10, 15], dtype='float32')
2164-
x_transposed = layers.transpose(input=x, perm=[1, 0, 2])
2164+
x_transposed = layers.transpose(x, perm=[1, 0, 2])
21652165
"""
21662166

2167-
if len(perm) != len(input.shape):
2167+
if len(perm) != len(x.shape):
21682168
raise ValueError(
21692169
"Input(perm) is the permutation of dimensions of Input(input). "
21702170
"It's length shoud be equal to Input(input)'s rank.")
21712171

21722172
helper = LayerHelper('transpose', **locals())
2173-
out = helper.create_tmp_variable(helper.input_dtype())
2173+
out = helper.create_tmp_variable(x.dtype)
21742174
helper.append_op(
21752175
type='transpose',
2176-
inputs={'X': [input]},
2176+
inputs={'X': [x]},
21772177
outputs={'Out': [out]},
21782178
attrs={'axis': perm})
21792179
return out

python/paddle/v2/fluid/layers/ops.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -45,10 +45,20 @@
4545
]
4646

4747
__all__ = [
48-
'mean', 'mul', 'reshape', 'scale', 'transpose',
49-
'sigmoid_cross_entropy_with_logits', 'elementwise_add', 'elementwise_div',
50-
'elementwise_sub', 'elementwise_mul', 'elementwise_max', 'elementwise_min',
51-
'clip', 'clip_by_norm', 'sequence_softmax'
48+
'mean',
49+
'mul',
50+
'reshape',
51+
'scale',
52+
'sigmoid_cross_entropy_with_logits',
53+
'elementwise_add',
54+
'elementwise_div',
55+
'elementwise_sub',
56+
'elementwise_mul',
57+
'elementwise_max',
58+
'elementwise_min',
59+
'clip',
60+
'clip_by_norm',
61+
'sequence_softmax',
5262
] + __activations__
5363

5464
for _OP in set(__all__):

python/paddle/v2/fluid/tests/book/test_understand_sentiment_lstm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,13 +65,13 @@ def lstm_net(dict_dim, class_dim=2, emb_dim=32, seq_len=80, batch_size=50):
6565

6666
emb = fluid.layers.embedding(input=data, size=[dict_dim, emb_dim])
6767
emb = fluid.layers.reshape(x=emb, shape=[batch_size, seq_len, emb_dim])
68-
emb = fluid.layers.transpose(x=emb, axis=[1, 0, 2])
68+
emb = fluid.layers.transpose(x=emb, perm=[1, 0, 2])
6969

7070
c_pre_init = fluid.layers.fill_constant(
7171
dtype=emb.dtype, shape=[batch_size, emb_dim], value=0.0)
7272
c_pre_init.stop_gradient = False
7373
layer_1_out = lstm(emb, c_pre_init=c_pre_init, hidden_dim=emb_dim)
74-
layer_1_out = fluid.layers.transpose(x=layer_1_out, axis=[1, 0, 2])
74+
layer_1_out = fluid.layers.transpose(x=layer_1_out, perm=[1, 0, 2])
7575

7676
prediction = fluid.layers.fc(input=layer_1_out,
7777
size=class_dim,

0 commit comments

Comments
 (0)