style

huggingface · Jun 25, 2024 · 25702ff · 25702ff
1 parent febcf1a
commit 25702ff
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -939,9 +939,7 @@ def test_load_model_from_hub_private(self):
         if token is None:
             self.skipTest("Test requires a token for fxmartyclone in the environment variable `HF_HUB_READ_TOKEN`.")
 
-        model = ORTModelForCustomTasks.from_pretrained(
-            "optimum-internal-testing/tiny-random-phi-private", token=token
-        )
+        model = ORTModelForCustomTasks.from_pretrained("optimum-internal-testing/tiny-random-phi-private", token=token)
 
         self.assertIsInstance(model.model, onnxruntime.InferenceSession)
         self.assertIsInstance(model.config, PretrainedConfig)