Fix CI

keras-team · Nov 3, 2023 · c1ec2aa · c1ec2aa
1 parent 149d1e1
commit c1ec2aa
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/keras/utils/audio_dataset_utils_test.py b/keras/utils/audio_dataset_utils_test.py
@@ -386,7 +386,7 @@ def test_audio_dataset_from_directory_errors(self):
             ValueError, "`class_names` passed did not match"
         ):
             _ = audio_dataset_utils.audio_dataset_from_directory(
-                directory, class_names=["class_0", "class_2"]
+                directory, class_names=["class_0", "wrong_class"]
             )
 
         with self.assertRaisesRegex(ValueError, "there must be exactly 2"):

diff --git a/keras/utils/image_dataset_utils_test.py b/keras/utils/image_dataset_utils_test.py
@@ -370,7 +370,7 @@ def test_image_dataset_from_directory_errors(self):
             ValueError, "`class_names` passed did not match"
         ):
             _ = image_dataset_utils.image_dataset_from_directory(
-                directory, class_names=["class_0", "class_2"]
+                directory, class_names=["class_0", "wrong_class"]
             )
 
         with self.assertRaisesRegex(ValueError, "there must be exactly 2"):

diff --git a/keras/utils/text_dataset_utils_test.py b/keras/utils/text_dataset_utils_test.py
@@ -253,7 +253,7 @@ def test_text_dataset_from_directory_errors(self):
             ValueError, "`class_names` passed did not match"
         ):
             _ = text_dataset_utils.text_dataset_from_directory(
-                directory, class_names=["class_0", "class_2"]
+                directory, class_names=["class_0", "wrong_class"]
             )
 
         with self.assertRaisesRegex(ValueError, "there must be exactly 2"):