Skip to content

Commit 6db5ce5

Browse files
committed
fix tests
1 parent e97f12e commit 6db5ce5

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tensorflow_datasets/image/imagewang_test.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,26 +12,26 @@ class ImagewangFullSizeTest(testing.DatasetBuilderTestCase):
1212
DATASET_CLASS = imagewang.Imagewang
1313
BUILDER_CONFIG_NAMES_TO_TEST = ['full-size']
1414
SPLITS = {
15-
"train": 4,
16-
"test": 4,
15+
"train": 1,
16+
"validation": 1,
1717
}
1818

1919

2020
class Imagewang320Test(testing.DatasetBuilderTestCase):
2121
DATASET_CLASS = imagewang.Imagewang
2222
BUILDER_CONFIG_NAMES_TO_TEST = ['320px']
2323
SPLITS = {
24-
"train": 4,
25-
"test": 4,
24+
"train": 1,
25+
"validation": 1,
2626
}
2727

2828

2929
class Imagewang160Test(testing.DatasetBuilderTestCase):
3030
DATASET_CLASS = imagewang.Imagewang
3131
BUILDER_CONFIG_NAMES_TO_TEST = ['160px']
3232
SPLITS = {
33-
"train": 4,
34-
"test": 4,
33+
"train": 1,
34+
"validation": 1,
3535
}
3636

3737

0 commit comments

Comments
 (0)