Merge pull request #237 from IAHispano/formatter-main

chore(format): run black on main
IAHispano · Jan 24, 2024 · e38be47 · e38be47
2 parents 87ec51e + 1794ba2
commit e38be47
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tabs/download/download.py b/tabs/download/download.py
@@ -29,9 +29,13 @@ def save_drop_model(dropbox):
             model_name = format_title(file_name.split(".pth")[0])
         else:
             if "v2" not in dropbox:
-                model_name = format_title(file_name.split("_nprobe_1_")[1].split("_v1")[0])
+                model_name = format_title(
+                    file_name.split("_nprobe_1_")[1].split("_v1")[0]
+                )
             else:
-                model_name = format_title(file_name.split("_nprobe_1_")[1].split("_v2")[0])
+                model_name = format_title(
+                    file_name.split("_nprobe_1_")[1].split("_v2")[0]
+                )
         model_path = os.path.join(now_dir, "logs", model_name)
         if not os.path.exists(model_path):
             os.makedirs(model_path)