style and quality

2020-02-07 08:58:06 +01:00
parent 961c69776f
commit c6c5c3fd4e
2 changed files with 6 additions and 2 deletions
--- a/src/transformers/modeling_utils.py
+++ b/src/transformers/modeling_utils.py
@@ -421,7 +421,9 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin):
                )
                archive_file = pretrained_model_name_or_path + ".index"
            else:
-                archive_file = hf_bucket_url(pretrained_model_name_or_path, postfix=(TF2_WEIGHTS_NAME if from_tf else WEIGHTS_NAME))
+                archive_file = hf_bucket_url(
+                    pretrained_model_name_or_path, postfix=(TF2_WEIGHTS_NAME if from_tf else WEIGHTS_NAME)
+                )

            # redirect to the cache, if necessary
            try: