Explicitly setting deep_weeds to be unpacked with ZIP.

PiperOrigin-RevId: 266752800
Anirban6393 · Sep 2, 2019 · 39771f7 · 39771f7
1 parent dda6584
commit 39771f7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tensorflow_datasets/image/deep_weeds.py b/tensorflow_datasets/image/deep_weeds.py
@@ -92,7 +92,11 @@ def _info(self):
 
   def _split_generators(self, dl_manager):
     """Define Splits."""
-    path = dl_manager.download_and_extract(_URL)
+    # The file is in ZIP format, but URL doesn't mention it.
+    path = dl_manager.download_and_extract(
+        tfds.download.Resource(
+            url=_URL,
+            extract_method=tfds.download.ExtractMethod.ZIP))
 
     return [
         tfds.core.SplitGenerator(