Fix duplicate entries caused by types in the Args: block

PiperOrigin-RevId: 530431744

Author: MarkDaoust

tools/tensorflow_docs/api_generator/parser.py

@@ -381,7 +381,11 @@ def __str__(self) -> str:
       r"""
       ^(\*?\*?'?"?     # Optional * to allow *args, **kwargs and quotes
           \w[\w.'"]*?  # words, dots and closing quotes
-          (?:[ ]?[\(\]][\(\[\w.\)\]]*?)? # maybe a space and some words in parens.
+          (?:          # non capturing
+              [ ]?     # maybe a space
+              \(       # a `(`
+              [\ \(\[\w.,\)\]]*? # word chars, dots or more open/close or space
+          )?           # all optional
       )\s*:\s          # Allow any whitespace around the colon.""",
       re.MULTILINE | re.VERBOSE,
   )
@@ -447,11 +451,26 @@ def split_string(cls, docstring: str):
       text = split.pop(0)
       items = _pairs(split)
 
+      items = list(cls._split_items(items))
+
       title_block = cls(title=title, text=text, items=items)
       parts.append(title_block)
 
     return parts
 
+  @classmethod
+  def _split_items(
+      cls, items: Iterable[Tuple[str, str]]
+  ) -> Iterable[Tuple[str, str]]:
+    """If there's a type in the name, move it to the top of the description."""
+    for name, value in items:
+      if '(' in name:
+        name, type_str = re.split(r' ?[\(]', name, maxsplit=1)
+        type_str = f"`{type_str.rstrip(')')}`"
+        value = f'{type_str}\n\n{value}'
+
+      yield name, value
+
 
 class DocstringInfo(typing.NamedTuple):
   brief: str

tools/tensorflow_docs/api_generator/parser_test.py

@@ -884,7 +884,8 @@ def test_title_block(self):
     block = docstring_parts[1]
     self.assertEqual('\nextra paragraph?\n', block.text)
     self.assertEqual('item', block.items[0][0])
-    self.assertEqual('item2 (int)', block.items[1][0])
+    self.assertEqual('item2', block.items[1][0])
+    self.assertStartsWith(block.items[1][1], '`int`')
     self.assertLen(block.items, 2)
 
   def test_strip_todos(self):