Skip to content

Commit a20f7f7

Browse files
committed
clean up for PR#1326
1 parent 0fd6207 commit a20f7f7

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

tensorflow_datasets/image/imagewang.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ class ImagewangConfig(tfds.core.BuilderConfig):
6666

6767
def __init__(self, size, **kwargs):
6868
super(ImagewangConfig, self).__init__(
69-
version=tfds.core.Version("0.1.0"), **kwargs)
69+
version=tfds.core.Version("2.0.0"), **kwargs)
7070
self.size = size
7171

7272

@@ -83,8 +83,8 @@ def _make_builder_configs():
8383

8484
class Imagewang(tfds.core.GeneratorBasedBuilder):
8585
""" Imagewang contains Imagenette and Imagewoof combined. """
86-
87-
VERSION = tfds.core.Version("0.1.0")
86+
87+
VERSION = tfds.core.Version("2.0.0")
8888
BUILDER_CONFIGS = _make_builder_configs()
8989

9090
def _info(self):
@@ -107,14 +107,12 @@ def _split_generators(self, dl_manager):
107107
if size in _SIZES:
108108
size_str = "" if size == "full-size" else "-" + size[:-2]
109109
url = os.path.join(_URL_PREFIX, "imagewang%s.tgz" % size_str)
110-
print(url)
111110
path = dl_manager.download_and_extract(url)
112111
train_path = os.path.join(path, _SIZE_TO_DIRNAME[size], "train")
113112
val_path = os.path.join(path, _SIZE_TO_DIRNAME[size], "val")
114113
else:
115-
raise ValueError("Size not implemented!")
114+
raise ValueError("size must be one of %s" % _SIZES)
116115

117-
print(train_path, 'train_path')
118116
return [
119117
tfds.core.SplitGenerator(
120118
name=tfds.Split.TRAIN,

0 commit comments

Comments
 (0)