diff --git a/utils/model_uploader/save_model_file_path_to_env.py b/utils/model_uploader/save_model_file_path_to_env.py index c0fe9cb1..537d5127 100644 --- 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 index f6a664e4..f1b8be4c 100644 --- 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(