SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Revert "add return_tensor parameter for feature extraction (#19257)" (#19680)

Browse Source 
This reverts commit 35bd089a24.
This commit is contained in:
Sylvain Gugger
2022-10-17 11:56:29 -04:00
committed by GitHub
parent bf0addc56e
commit f2ecb9eec4
2 changed files with 3 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/pipelines/test_pipelines_feature_extraction.py
View File

@@ -15,8 +15,6 @@
import unittest
import numpy as np
import tensorflow as tf
import torch
from transformers import (
FEATURE_EXTRACTOR_MAPPING,
@@ -135,22 +133,6 @@ class FeatureExtractionPipelineTests(unittest.TestCase, metaclass=PipelineTestCa
tokenize_kwargs=tokenize_kwargs,
)
@require_torch
def test_return_tensors_pt(self):
feature_extractor = pipeline(
task="feature-extraction", model="hf-internal-testing/tiny-random-distilbert", framework="pt"
)
outputs = feature_extractor("This is a test" * 100, return_tensors=True)
self.assertTrue(torch.is_tensor(outputs))
@require_tf
def test_return_tensors_tf(self):
feature_extractor = pipeline(
task="feature-extraction", model="hf-internal-testing/tiny-random-distilbert", framework="tf"
)
outputs = feature_extractor("This is a test" * 100, return_tensors=True)
self.assertTrue(tf.is_tensor(outputs))
def get_shape(self, input_, shape=None):
if shape is None:
shape = []