pyformat again

Author: markmcd

tools/tensorflow_docs/tools/nblint/style/gemini_cookbook.py

@@ -84,7 +84,8 @@ def copyright_check(args):
 @lint(
     message="Apache license cell is required",
     scope=Options.Scope.CODE,
-    cond=Options.Cond.ANY)
+    cond=Options.Cond.ANY,
+)
 def license_check(args):
   if license_re.search(args["cell_source"]):
     return True
@@ -126,8 +127,10 @@ def get_arg_or_fail(user_args, arg_name, arg_fmt):
     return user_args.get(arg_name)
   else:
     fail(
-        f"Requires user-argument '{arg_name}': nblint --arg={arg_name}:{arg_fmt} ...",
-        always_show=True)
+        f"Requires user-argument '{arg_name}': nblint"
+        f" --arg={arg_name}:{arg_fmt} ...",
+        always_show=True,
+    )
 
 
 def split_doc_path(filepath):
@@ -157,8 +160,8 @@ def split_doc_path(filepath):
   def split_path_on_dir(fp, dirname, offset=1):
     parts = fp.parts
     idx = parts.index(dirname)
-    docs_dir = pathlib.Path(*parts[idx:idx + offset])
-    rel_path = fp.relative_to(*parts[:idx + offset])
+    docs_dir = pathlib.Path(*parts[idx : idx + offset])
+    rel_path = fp.relative_to(*parts[: idx + offset])
     return docs_dir, rel_path
 
   if "site" in fp_full.parts:
@@ -180,7 +183,8 @@ def split_path_on_dir(fp, dirname, offset=1):
 @lint(
     message="Missing or malformed URL in Colab button.",
     scope=Options.Scope.TEXT,
-    cond=Options.Cond.ANY)
+    cond=Options.Cond.ANY,
+)
 def button_colab(args):
   """Test that the URL in the Colab button matches the file path."""
   cell_source = args["cell_source"]
@@ -201,13 +205,15 @@ def button_colab(args):
     fail(
         f"Colab button URL doesn't match: {this_url}",
         fix=fix.regex_between_groups_replace_all,
-        fix_args=[r"(href.*)http.*?(\\\".*colab_logo_32px.png)", this_url])
+        fix_args=[r"(href.*)http.*?(\\\".*colab_logo_32px.png)", this_url],
+    )
 
 
 @lint(
     message="Missing or malformed URL in Download button.",
     scope=Options.Scope.TEXT,
-    cond=Options.Cond.ANY)
+    cond=Options.Cond.ANY,
+)
 def button_download(args):
   """Test that the URL in the Download button matches the file path."""
   cell_source = args["cell_source"]
@@ -224,21 +230,24 @@ def button_download(args):
 
   this_url = urllib.parse.urljoin(
       "https://storage.googleapis.com",
-      str(f"tensorflow_docs/{repo_name}" / docs_dir / rel_path))
+      str(f"tensorflow_docs/{repo_name}" / docs_dir / rel_path),
+  )
 
   if is_button_cell_re.search(cell_source) and cell_source.find(this_url) != -1:
     return True
   else:
     fail(
         f"Download button URL doesn't match: {this_url}",
         fix=fix.regex_between_groups_replace_all,
-        fix_args=[r"(href.*)http.*?(\\\".*download_logo_32px.png)", this_url])
+        fix_args=[r"(href.*)http.*?(\\\".*download_logo_32px.png)", this_url],
+    )
 
 
 @lint(
     message="Missing or malformed URL in GitHub button.",
     scope=Options.Scope.TEXT,
-    cond=Options.Cond.ANY)
+    cond=Options.Cond.ANY,
+)
 def button_github(args):
   """Test that the URL in the GitHub button matches the file path."""
   cell_source = args["cell_source"]
@@ -259,13 +268,15 @@ def button_github(args):
     fail(
         f"GitHub button URL doesn't match: {this_url}",
         fix=fix.regex_between_groups_replace_all,
-        fix_args=[r"(href.*)http.*?(\\\".*GitHub-Mark-32px.png)", this_url])
+        fix_args=[r"(href.*)http.*?(\\\".*GitHub-Mark-32px.png)", this_url],
+    )
 
 
 @lint(
     message="Missing or malformed URL in 'View on' button.",
     scope=Options.Scope.TEXT,
-    cond=Options.Cond.ANY)
+    cond=Options.Cond.ANY,
+)
 def button_website(args):
   """Test that the website URL in the 'View on' button matches the file path.
 
@@ -310,7 +321,8 @@ def button_website(args):
 @lint(
     message="Missing or malformed URL in 'TFHub' button.",
     scope=Options.Scope.TEXT,
-    cond=Options.Cond.ANY)
+    cond=Options.Cond.ANY,
+)
 def button_hub(args):
   """Notebooks that mention tfhub.dev should have a TFHub button."""
   cell_source = args["cell_source"]
@@ -333,7 +345,8 @@ def button_hub(args):
 @lint(
     message="Remove extra buttons from TF 1.x docs.",
     scope=Options.Scope.TEXT,
-    cond=Options.Cond.ALL)
+    cond=Options.Cond.ALL,
+)
 def button_r1_extra(args):
   """The r1/ docs should not have website or download buttons."""
   cell_source = args["cell_source"]
@@ -353,10 +366,13 @@ def button_r1_extra(args):
     base_url = "https://www.tensorflow.org/"
 
   # Look for button URLs that shouldn't be there..
-  if (re.search(f"{base_url}/(?!images)", cell_source) or
-      cell_source.find(download_url) != -1):
+  if (
+      re.search(f"{base_url}/(?!images)", cell_source)
+      or cell_source.find(download_url) != -1
+  ):
     fail(
-        "Remove the 'View on' and 'Download notebook' buttons since r1/ docs are not published."
+        "Remove the 'View on' and 'Download notebook' buttons since r1/ docs"
+        " are not published."
     )
   else:
     return True