Skip private model loading for external contributors (#2130)

echarlaix · web-flow · commit 34b3d8bdfebe · 2024-12-19T16:21:13.000+01:00
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -974,7 +974,7 @@ def test_stable_diffusion_model_on_rocm_ep_str(self):
     def test_load_model_from_hub_private(self):
         token = os.environ.get("HF_HUB_READ_TOKEN", None)
 
-        if token is None:
+        if not token:
             self.skipTest(
                 "Test requires a read access token for optimum-internal-testing in the environment variable `HF_HUB_READ_TOKEN`."
             )

Original file line number	Diff line number	Diff line change
`@@ -974,7 +974,7 @@ def test_stable_diffusion_model_on_rocm_ep_str(self):`
`974`	`974`	`def test_load_model_from_hub_private(self):`
`975`	`975`	`token = os.environ.get("HF_HUB_READ_TOKEN", None)`
`976`	`976`
`977`		`- if token is None:`
	`977`	`+ if not token:`
`978`	`978`	`self.skipTest(`
`979`	`979`	"Test requires a read access token for optimum-internal-testing in the environment variable `HF_HUB_READ_TOKEN`."
`980`	`980`	`)`