fix tests

Author: yonigozlan

tests/models/dac/test_feature_extraction_dac.py

@@ -207,7 +207,7 @@ def test_truncation_and_padding(self):
         # force no pad
         with self.assertRaisesRegex(
             ValueError,
-            "^Unable to create tensor, you should probably activate padding with 'padding=True' to have batched tensors with the same length.$",
+            r"Unable to convert output[\s\S]*padding=True",
         ):
             truncated_outputs = feature_extractor(input_audio, padding=False, return_tensors="pt").input_values
 

tests/models/dia/test_feature_extraction_dia.py

@@ -223,7 +223,7 @@ def test_truncation_and_padding(self):
         # force no pad
         with self.assertRaisesRegex(
             ValueError,
-            "^Unable to create tensor, you should probably activate padding with 'padding=True' to have batched tensors with the same length.$",
+            r"Unable to convert output[\s\S]*padding=True",
         ):
             truncated_outputs = feature_extractor(input_audio, padding=False, return_tensors="pt").input_values
 

tests/models/encodec/test_feature_extraction_encodec.py

@@ -221,7 +221,7 @@ def test_truncation_and_padding(self):
         # force no pad
         with self.assertRaisesRegex(
             ValueError,
-            "^Unable to create tensor, you should probably activate padding with 'padding=True' to have batched tensors with the same length.$",
+            r"Unable to convert output[\s\S]*padding=True",
         ):
             truncated_outputs = feature_extractor(input_audio, padding=False, return_tensors="pt").input_values
 
@@ -232,7 +232,7 @@ def test_truncation_and_padding(self):
         feature_extractor.chunk_length_s = None
         with self.assertRaisesRegex(
             ValueError,
-            "^Unable to create tensor, you should probably activate padding with 'padding=True' to have batched tensors with the same length.$",
+            r"Unable to convert output[\s\S]*padding=True",
         ):
             truncated_outputs = feature_extractor(input_audio, padding=False, return_tensors="pt").input_values
 
@@ -244,7 +244,7 @@ def test_truncation_and_padding(self):
         feature_extractor.overlap = None
         with self.assertRaisesRegex(
             ValueError,
-            "^Unable to create tensor, you should probably activate padding with 'padding=True' to have batched tensors with the same length.$",
+            r"Unable to convert output[\s\S]*padding=True",
         ):
             truncated_outputs = feature_extractor(input_audio, padding=False, return_tensors="pt").input_values