Correct linting

Signed-off-by: Thanawan Atchariyachanvanit <[email protected]>
thanawan-atc · Aug 17, 2023 · 68198c8 · 68198c8
1 parent 38b3e4f
commit 68198c8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/utils/model_uploader/save_model_file_path_to_env.py b/utils/model_uploader/save_model_file_path_to_env.py
@@ -52,9 +52,7 @@ def get_model_file_path(
     model_type, model_name = model_id.split("/")
     model_format = model_format.lower()
     model_dirname = f"{model_folder}{model_name}/{model_version}/{model_format}"
-    model_filename = (
-        f"{model_type}_{model_name}-{model_version}-{model_format}.zip"
-    )
+    model_filename = f"{model_type}_{model_name}-{model_version}-{model_format}.zip"
     model_file_path = model_dirname + "/" + model_filename
     return model_file_path
 

diff --git a/utils/model_uploader/update_pretrained_model_listing.py b/utils/model_uploader/update_pretrained_model_listing.py
@@ -128,12 +128,10 @@ def create_new_pretrained_model_listing(
         type=str,
         help="Name of the local folder that stores copies of config files from S3",
     )
-    
+
     args = parser.parse_args()
 
-    if not args.config_paths_txt_filename.endswith(
-        ".txt"
-    ):
+    if not args.config_paths_txt_filename.endswith(".txt"):
         assert False, "Invalid arguments"
 
     create_new_pretrained_model_listing(