Skip to content
This repository was archived by the owner on Dec 19, 2023. It is now read-only.

Commit 0707e18

Browse files
committed
possible fix for rce in eval()
1 parent 39db6a9 commit 0707e18

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensorlayer/files/utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -238,18 +238,18 @@ def eval_layer(layer_kwargs):
238238
for layer_graph in layers:
239239
ret_layer.append(eval_layer(layer_graph))
240240
args['layers'] = ret_layer
241-
return eval('tl.layers.' + layer_class)(**args)
241+
return ast.literal_eval('tl.layers.' + layer_class)(**args)
242242
elif layer_type == "modellayer":
243243
M = static_graph2net(args['model'])
244244
args['model'] = M
245-
return eval('tl.layers.' + layer_class)(**args)
245+
return ast.literal_eval('tl.layers.' + layer_class)(**args)
246246
elif layer_type == "keraslayer":
247247
M = load_keras_model(args['fn'])
248248
input_shape = args.pop('keras_input_shape')
249249
_ = M(np.random.random(input_shape).astype(np.float32))
250250
args['fn'] = M
251251
args['fn_weights'] = M.trainable_variables
252-
return eval('tl.layers.' + layer_class)(**args)
252+
return ast.literal_eval('tl.layers.' + layer_class)(**args)
253253
else:
254254
raise RuntimeError("Unknown layer type.")
255255

0 commit comments

Comments
 (0)