Fix string comparison (#1014)

Fix string comparison in validate_initializer_name Signed-off-by: Chin Huang <[email protected]>
onnx · Mar 17, 2022 · e2a8a71 · e2a8a71
1 parent f6bebcc
commit e2a8a71
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnx_tf/backend.py b/onnx_tf/backend.py
@@ -292,8 +292,8 @@ def tensor2list(onnx_tensor):
       return numpy_helper.to_array(onnx_tensor).flatten().tolist()
 
     def validate_initializer_name(name):
-      # Prepend a unique suffix if leading charater is "_"
-      name = get_unique_suffix() + name if name[0] is "_" else name
+      # Prepend a unique suffix if leading character is "_"
+      name = get_unique_suffix() + name if name[0] == "_" else name
 
       # Replace ":" with "_tf_" and append a unique suffix for
       # traceability