Skip to content

Commit

Permalink
removed the return type test for the feature extractor
Browse files Browse the repository at this point in the history
  • Loading branch information
FrancescoSaverioZuppichini committed Feb 23, 2022
1 parent b564929 commit e5d8cc0
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions tests/test_feature_extraction_maskformer.py
Expand Up @@ -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
Expand Down

0 comments on commit e5d8cc0

Please sign in to comment.