fix test

openAGI · Mar 5, 2025 · 6f7ef4a · 6f7ef4a
1 parent cbc3abc
commit 6f7ef4a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_parser.py b/tests/test_parser.py
@@ -136,4 +136,4 @@ def test_parser(self):
     batch = next(iter(dataset))
     self.assertEqual(list(batch.keys()), ['image', 'label'])
     self.assertEqual(batch['image'].shape, [1, 366, 500, 3])
-    self.assertEqual(batch['label'].shape, [1, 366, 500, 3])
+    self.assertEqual(batch['label'].shape, [1, 366, 500, 1])