Support spaces in license headers

When running `pyink` (black) on a notebook with a license, it adds a space. We should permit this.
tensorflow · May 3, 2024 · 0675307 · 0675307
1 parent 3eac15b
commit 0675307
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/tensorflow_docs/tools/nbfmt/__main__.py b/tools/tensorflow_docs/tools/nbfmt/__main__.py
@@ -237,7 +237,7 @@ def update_license_cells(data: Dict[str, Any]) -> None:
     data: object representing a parsed JSON notebook.
   """
   # This pattern in Apache and MIT license boilerplate.
-  license_re = re.compile(r"#@title.*License")
+  license_re = re.compile(r"#\s?@title.*License")
 
   for idx, cell in enumerate(data["cells"]):
     src_text = "".join(cell["source"])