Retry parsing a module that has invalid type comments

It is possible for a module to use comments that might be interpreted
as type comments by the `ast` library. We do not want to completely crash on those
invalid type comments.

Close #708

1 files changed, 25 insertions, 0 deletions

diff --git a/tests/unittest_builder.py b/tests/unittest_builder.py
index 0f1a470f..75557761 100644
--- a/tests/unittest_builder.py
+++ b/tests/unittest_builder.py
@@ -692,5 +692,30 @@ def test_module_build_dunder_file():
     assert module.path[0] == collections.__file__
 
 
+def test_parse_module_with_invalid_type_comments_does_not_crash():
+    node = builder.parse(
+        """
+    # op {
+    #   name: "AssignAddVariableOp"
+    #   input_arg {
+    #     name: "resource"
+    #     type: DT_RESOURCE
+    #   }
+    #   input_arg {
+    #     name: "value"
+    #     type_attr: "dtype"
+    #   }
+    #   attr {
+    #     name: "dtype"
+    #     type: "type"
+    #   }
+    #   is_stateful: true
+    # }
+    a, b = 2
+    """
+    )
+    assert isinstance(node, nodes.Module)
+
+
 if __name__ == "__main__":
     unittest.main()