string attributes bug fixes

PINTO0309 · Jun 7, 2022 · 3e15658 · 3e15658
1 parent bc314df
commit 3e15658
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/sog4onnx/__init__.py b/sog4onnx/__init__.py
@@ -1,3 +1,3 @@
 from sog4onnx.onnx_operation_generator import generate, main
 
-__version__ = '1.0.11'
+__version__ = '1.0.12'
diff --git a/sog4onnx/onnx_operation_generator.py b/sog4onnx/onnx_operation_generator.py
@@ -360,7 +360,12 @@ def main():
             # parse
             attr_name = attribute[0]
             attr_type = attribute[1]
-            attr_value = ast.literal_eval(attribute[2])
+            if attr_type == 'string':
+                attr_type = 'str'
+            if attr_type != 'str':
+                attr_value = ast.literal_eval(attribute[2])
+            else:
+                attr_value = attribute[2]
 
             # dtype check
             if attr_type not in AVAILABLE_DTYPES: