format code

Author: longendu

aiutil/filesystem.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Filesystem related util functions."""
+
 from typing import Iterable, Callable
 import itertools
 import os
@@ -548,9 +549,10 @@ def dedup_header(file: str | Path, output: str | Path = "") -> None:
     :param output: The path of the output file.
         If empty, then output to the standard output.
     """
-    with open(file, "rb") as fin, (
-        open(output, "wb") if output else sys.stdout.buffer
-    ) as fout:
+    with (
+        open(file, "rb") as fin,
+        open(output, "wb") if output else sys.stdout.buffer as fout,
+    ):
         header = fin.readline()
         fout.write(header)
         for line in fin:
@@ -606,9 +608,10 @@ def prune_json(input: str | Path, output: str | Path = ""):
         else:
             output = input.with_name(input.stem + "_prune.json")
     skip = False
-    with input.open("r", encoding="utf-8") as fin, output.open(
-        "w", encoding="utf-8"
-    ) as fout:
+    with (
+        input.open("r", encoding="utf-8") as fin,
+        output.open("w", encoding="utf-8") as fout,
+    ):
         for line in fin:
             line = line.strip()
             if line == '"value_counts": {':

aiutil/hadoop/kerberos.py

@@ -5,6 +5,7 @@
 If an user authenticate without specifying password,
 the saved password is used so that users do not have to type in password to authenticate every time.
 """
+
 from typing import Any
 import os
 from pathlib import Path
@@ -81,7 +82,7 @@ def authenticate(password: str, email: dict[str, str], user: str = "") -> None:
     :param user: User name. If empty, the current user name is used.
     """
     SUBJECT = "kinit: authentication {}"
-    MSG = f'kinit ({PID}): authentication on {HOST} ({HOST_IP}) {"{}"} at {datetime.datetime.now()}'
+    MSG = f"kinit ({PID}): authentication on {HOST} ({HOST_IP}) {'{}'} at {datetime.datetime.now()}"
     try:
         process = sp.run(
             ["/usr/bin/kinit", user if user else USER],

aiutil/hadoop/pyspark_submit.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 # encoding: utf-8
 """A module makes it easy to run Scala/Python Spark job."""
+
 from typing import Callable, Any, Iterable
 import os
 import sys

aiutil/notebook/search.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Search Jupyter/Lab notebooks."""
+
 from typing import TypeAlias, Sequence
 import json
 from pathlib import Path

aiutil/notebook/util.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Jupyter/Lab notebooks related utils."""
+
 from pathlib import Path
 import subprocess as sp
 import itertools as it