make fixup

huggingface · Aug 27, 2024 · 086354a · 086354a
1 parent 006b8c6
commit 086354a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/models/auto/test_modeling_auto.py b/tests/models/auto/test_modeling_auto.py
@@ -21,6 +21,7 @@
 from pathlib import Path
 
 import pytest
+from huggingface_hub import Repository
 
 import transformers
 from transformers import BertConfig, GPT2Model, is_safetensors_available, is_torch_available
@@ -32,7 +33,6 @@
     require_torch,
     slow,
 )
-from huggingface_hub import Repository
 
 from ..bert.test_modeling_bert import BertModelTester
 
@@ -538,4 +538,4 @@ def test_dynamic_saving_from_local_repo(self):
             model.save_pretrained(tmp_dir_out)
             _ = AutoModelForCausalLM.from_pretrained(tmp_dir_out, trust_remote_code=True)
             self.assertTrue((Path(tmp_dir_out) / "modeling_fake_custom.py").is_file())
-            self.assertTrue((Path(tmp_dir_out) / "configuration_fake_custom.py").is_file())
+            self.assertTrue((Path(tmp_dir_out) / "configuration_fake_custom.py").is_file())
diff --git a/tests/pipelines/test_pipelines_common.py b/tests/pipelines/test_pipelines_common.py
@@ -36,7 +36,7 @@
     TFAutoModelForSequenceClassification,
     pipeline,
 )
-from transformers.pipelines import PIPELINE_REGISTRY, TextGenerationPipeline, get_task
+from transformers.pipelines import PIPELINE_REGISTRY, get_task
 from transformers.pipelines.base import Pipeline, _pad
 from transformers.testing_utils import (
     TOKEN,