@@ -66,7 +66,7 @@ class ImagewangConfig(tfds.core.BuilderConfig):
66
66
67
67
def __init__ (self , size , ** kwargs ):
68
68
super (ImagewangConfig , self ).__init__ (
69
- version = tfds .core .Version ("0.1 .0" ), ** kwargs )
69
+ version = tfds .core .Version ("2.0 .0" ), ** kwargs )
70
70
self .size = size
71
71
72
72
@@ -83,8 +83,8 @@ def _make_builder_configs():
83
83
84
84
class Imagewang (tfds .core .GeneratorBasedBuilder ):
85
85
""" Imagewang contains Imagenette and Imagewoof combined. """
86
-
87
- VERSION = tfds .core .Version ("0.1 .0" )
86
+
87
+ VERSION = tfds .core .Version ("2.0 .0" )
88
88
BUILDER_CONFIGS = _make_builder_configs ()
89
89
90
90
def _info (self ):
@@ -107,14 +107,12 @@ def _split_generators(self, dl_manager):
107
107
if size in _SIZES :
108
108
size_str = "" if size == "full-size" else "-" + size [:- 2 ]
109
109
url = os .path .join (_URL_PREFIX , "imagewang%s.tgz" % size_str )
110
- print (url )
111
110
path = dl_manager .download_and_extract (url )
112
111
train_path = os .path .join (path , _SIZE_TO_DIRNAME [size ], "train" )
113
112
val_path = os .path .join (path , _SIZE_TO_DIRNAME [size ], "val" )
114
113
else :
115
- raise ValueError ("Size not implemented!" )
114
+ raise ValueError ("size must be one of %s" % _SIZES )
116
115
117
- print (train_path , 'train_path' )
118
116
return [
119
117
tfds .core .SplitGenerator (
120
118
name = tfds .Split .TRAIN ,
0 commit comments