From e5d8cc0f193e1c8c16c0ac1efd1a266749c6d01d Mon Sep 17 00:00:00 2001 From: fra Date: Wed, 23 Feb 2022 13:13:25 +0100 Subject: [PATCH] removed the return type test for the feature extractor --- tests/test_feature_extraction_maskformer.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/tests/test_feature_extraction_maskformer.py b/tests/test_feature_extraction_maskformer.py index 33bc92c342bb2f..e5ff4277a9b9e4 100644 --- a/tests/test_feature_extraction_maskformer.py +++ b/tests/test_feature_extraction_maskformer.py @@ -284,14 +284,6 @@ def test_with_size_divisibility(self): self.assertTrue((pixel_values.shape[-1] % size_divisibility) == 0) self.assertTrue((pixel_values.shape[-2] % size_divisibility) == 0) - def test_return_tensors(self): - faulty_tensor_types = ["tf", "jax", "np"] - for tensor_type in faulty_tensor_types: - feature_extractor = self.feature_extraction_class(**self.feat_extract_dict) - with pytest.raises(Exception): - # this should fail one way or another - feature_extractor([np.random.rand(3, 100, 100)], return_tensors=tensor_type) - def test_call_with_numpy_annotations(self): num_classes = 8 batch_size = self.feature_extract_tester.batch_size