eval -> ast.literal_eval

PINTO0309 · May 15, 2022 · 7c55843 · 7c55843
1 parent 700ef70
commit 7c55843
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/scc4onnx/__init__.py b/scc4onnx/__init__.py
@@ -1,3 +1,3 @@
 from scc4onnx.onnx_input_order_converter import order_conversion, main
 
-__version__ = '1.0.2'
+__version__ = '1.0.3'
diff --git a/scc4onnx/onnx_input_order_converter.py b/scc4onnx/onnx_input_order_converter.py
@@ -2,6 +2,7 @@
 
 import os
 import sys
+import ast
 import traceback
 from argparse import ArgumentParser
 import onnx
@@ -507,7 +508,7 @@ def main():
     input_op_names_and_order_dims_tmp = None
     if input_op_names_and_order_dims:
         input_op_names_and_order_dims_tmp = {
-            input_op_names_and_order_dim[0]: eval(input_op_names_and_order_dim[1]) for input_op_names_and_order_dim in input_op_names_and_order_dims
+            input_op_names_and_order_dim[0]: ast.literal_eval(input_op_names_and_order_dim[1]) for input_op_names_and_order_dim in input_op_names_and_order_dims
         }
 
     # channel_change_inputs