Fix transfomers import in the evaluator (huggingface#291)

* Fix transfomers import in the evaluator * Fix imports
sywangyi · Sep 14, 2022 · abccc58 · abccc58
1 parent eef73da
commit abccc58
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/evaluate/evaluator/base.py b/src/evaluate/evaluator/base.py
@@ -18,7 +18,6 @@
 
 # Lint as: python3
 from datasets import Dataset, load_dataset
-from transformers import Pipeline
 
 from evaluate.evaluator.utils import choose_split
 
@@ -32,7 +31,7 @@
 
 try:
     import transformers
-    from transformers import pipeline
+    from transformers import Pipeline, pipeline
 
     TRANSFORMERS_AVAILABLE = True
 except ImportError: