Fix gated test (#2993)

* Fix gated test

* Clean

* Finally, adjust test

src/accelerate/commands/estimate.py

@@ -38,7 +38,7 @@ def verify_on_hub(repo: str, token: str = None):
     "Verifies that the model is on the hub and returns the model info."
     try:
         return model_info(repo, token=token)
-    except GatedRepoError:
+    except (OSError, GatedRepoError):
         return "gated"
     except RepositoryNotFoundError:
         return "repo"

tests/test_cli.py

@@ -435,7 +435,10 @@ def test_no_metadata(self):
             estimate_command(args)
 
     def test_gated(self):
-        with self.assertRaises(GatedRepoError, msg="Repo for model `meta-llama/Llama-2-7b-hf` is gated"):
+        with self.assertRaises(
+            (GatedRepoError, EnvironmentError),
+            msg="Repo for model `meta-llama/Llama-2-7b-hf` is gated or environment error occurred",
+        ):
             args = self.parser.parse_args(["meta-llama/Llama-2-7b-hf"])
             with patch_environment(hf_hub_disable_implicit_token="1"):
                 estimate_command(args)