From 25702ff3427c5886a972e48ac276ce44ca83aba7 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Tue, 25 Jun 2024 14:17:45 +0200 Subject: [PATCH] style --- tests/onnxruntime/test_modeling.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py index f09a190531f..e15a7e25d09 100644 --- 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)