Skip to content

GitLab

Unverified Commit 31728292 authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Fix the real failing test

parent 22888d30
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/models/wav2vec2/test_modeling_tf_wav2vec2.py
View file @ 31728292
...@@ -518,6 +518,7 @@ class TFWav2Vec2RobustModelTest(TFModelTesterMixin, unittest.TestCase): ...@@ -518,6 +518,7 @@ class TFWav2Vec2RobustModelTest(TFModelTesterMixin, unittest.TestCase):
self.assertIsNotNone(model) self.assertIsNotNone(model)
# We override here as passing a full batch of 13 samples results in OOM errors for CTC # We override here as passing a full batch of 13 samples results in OOM errors for CTC
@unittest.skip("Fix me!")
def test_dataset_conversion(self): def test_dataset_conversion(self):
default_batch_size = self.model_tester.batch_size default_batch_size = self.model_tester.batch_size
self.model_tester.batch_size = 2 self.model_tester.batch_size = 2
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment