reformat changed code

Author: Winter Deng

docs/cli/classifier.py

@@ -11,27 +11,37 @@
 
 import subprocess
 
+
 def load_exclude_authors(file_path):
     path = Path(file_path)
     if not path.exists():
         return []
     with open(path, "r", encoding="utf-8") as f:
         return [line.strip() for line in f if line.strip()]
 
+
 def get_last_commit_excluding(exclude_file):
     exclude_authors = set(load_exclude_authors(exclude_file))
-    
-    logs = subprocess.check_output([
-        "git", "log",
-        "--pretty=format:%H|%ae",  # commit hash | author email
-    ]).decode("utf-8").splitlines()
-    
+
+    logs = (
+        subprocess.check_output(
+            [
+                "git",
+                "log",
+                "--pretty=format:%H|%ae",  # commit hash | author email
+            ]
+        )
+        .decode("utf-8")
+        .splitlines()
+    )
+
     for line in logs:
         commit, email = line.split("|", 1)
         if email not in exclude_authors:
             return commit
     return None
 
+
 def classify_file_category(path):
 
     relative_path = Path(path).relative_to(lib_path)
@@ -52,12 +62,12 @@ def fetch_option_flags(flags):
 
     for flag in flags:
         flag_list.append(
-                {
-                    "name": flag["name"].replace("\\", ""),
-                    "instruction": flag["name"].split("-")[-1],
-                    "description": flag["description"]
-                }
-            )
+            {
+                "name": flag["name"].replace("\\", ""),
+                "instruction": flag["name"].split("-")[-1],
+                "description": flag["description"],
+            }
+        )
 
     return flag_list
 
@@ -66,15 +76,15 @@ def fetch_all_files():
     main_files = [
         os.path.join(lib_path, "main.py"),
         os.path.join(lib_path, "linear_trainer.py"),
-        os.path.join(lib_path, "torch_trainer.py")
+        os.path.join(lib_path, "torch_trainer.py"),
     ]
     lib_files = glob.glob(os.path.join(lib_path, "libmultilabel/**/*.py"), recursive=True)
     file_set = set(map(os.path.abspath, main_files + lib_files))
     return file_set
 
 
 def find_config_usages_in_file(file_path, allowed_keys):
-    pattern = re.compile(r'\bconfig\.([a-zA-Z_][a-zA-Z0-9_]*)')
+    pattern = re.compile(r"\bconfig\.([a-zA-Z_][a-zA-Z0-9_]*)")
     detailed_results = {}
     try:
         with open(file_path, "r", encoding="utf-8") as f:
@@ -114,7 +124,7 @@ def move_duplicates_together(data, keep):
     duplicates = set()
 
     for i, key1 in enumerate(all_keys):
-        for key2 in all_keys[i+1:]:
+        for key2 in all_keys[i + 1 :]:
             duplicates |= data[key1] & data[key2]
 
     data[keep] |= duplicates
@@ -136,7 +146,7 @@ def classify(raw_flags):
     collected = {}
 
     for file_path in file_set:
-        detailed_results = find_config_usages_in_file(file_path, allowed_keys)        
+        detailed_results = find_config_usages_in_file(file_path, allowed_keys)
         if detailed_results:
             usage_map[file_path] = set(detailed_results.keys())
             for k, v in detailed_results.items():
@@ -163,7 +173,9 @@ def classify(raw_flags):
         if flag["category"] not in result:
             result[flag["category"]] = []
 
-        result[flag["category"]].append({"name": flag["name"].replace("--", r"\-\-"), "description": flag["description"]})
+        result[flag["category"]].append(
+            {"name": flag["name"].replace("--", r"\-\-"), "description": flag["description"]}
+        )
 
     result["details"] = []
     for k, v in collected.items():
@@ -172,4 +184,4 @@ def classify(raw_flags):
             for i in v[1:]:
                 result["details"].append({"name": "", "file": i["file"], "location": ", ".join(i["lines"])})
 
-    return result
+    return result

docs/cli/genflags.py

@@ -6,6 +6,7 @@
 import main
 from classifier import classify
 
+
 class FakeParser(dict):
     def __init__(self):
         self.flags = []
@@ -32,9 +33,11 @@ def add_argument(
 
 classified = classify(parser.flags)
 
+
 def width_title(key, title):
     return max(map(lambda f: len(f[key]), classified[title]))
 
+
 def print_table(title, flags, intro):
     print()
     print(intro)
@@ -51,21 +54,22 @@ def print_table(title, flags, intro):
     print("=" * wn, "=" * wd)
     print()
 
+
 print_table(
     "general",
     classified["general"],
     intro="**General options**:\n\
-Common configurations shared across both linear and neural network trainers."
+Common configurations shared across both linear and neural network trainers.",
 )
 print_table(
     "linear",
     classified["linear"],
     intro="**Linear options**:\n\
-Configurations specific to linear trainer."
+Configurations specific to linear trainer.",
 )
 print_table(
     "nn",
     classified["nn"],
     intro="**Neural network options**:\n\
-Configurations specific to torch (neural networks) trainer."
+Configurations specific to torch (neural networks) trainer.",
 )

main.py

@@ -41,7 +41,11 @@ def add_all_arguments(parser):
     parser.add_argument("--checkpoint_path", help="The checkpoint to warm-up with (default: %(default)s)")
 
     # data
-    parser.add_argument("--data_name", default="unnamed_data", help="Dataset name for generating the output directory (default: %(default)s)")
+    parser.add_argument(
+        "--data_name",
+        default="unnamed_data",
+        help="Dataset name for generating the output directory (default: %(default)s)",
+    )
     parser.add_argument("--training_file", help="Path to training data (default: %(default)s)")
     parser.add_argument("--val_file", help="Path to validation data (default: %(default)s)")
     parser.add_argument("--test_file", help="Path to test data (default: %(default)s)")
@@ -104,7 +108,9 @@ def add_all_arguments(parser):
     # pretrained vocab / embeddings
     parser.add_argument("--vocab_file", type=str, help="Path to a file holding vocabuaries (default: %(default)s)")
     parser.add_argument(
-        "--embed_file", type=str, help="Path to a file holding pre-trained embeddings or the name of the pretrained GloVe embedding (default: %(default)s)"
+        "--embed_file",
+        type=str,
+        help="Path to a file holding pre-trained embeddings or the name of the pretrained GloVe embedding (default: %(default)s)",
     )
 
     # train
@@ -235,7 +241,10 @@ def add_all_arguments(parser):
         "--tree_max_depth", type=int, default=10, help="Maximum depth of the tree (default: %(default)s)"
     )
     parser.add_argument(
-        "--tree_ensemble_models", type=int, default=1, help="Number of models in the tree ensemble (default: %(default)s)"
+        "--tree_ensemble_models",
+        type=int,
+        default=1,
+        help="Number of models in the tree ensemble (default: %(default)s)",
     )
     parser.add_argument(
         "--beam_width",