Add uppercased versions of image file extensions for automatic module inference (huggingface#4515)

mariosasko · khushmeeet · commit 8ed5c2ce39f5 · 2022-06-22T21:34:08.000-04:00
diff --git a/src/datasets/packaged_modules/__init__.py b/src/datasets/packaged_modules/__init__.py
@@ -43,3 +43,4 @@ def _hash_python_lines(lines: List[str]) -> str:
     "txt": ("text", {}),
 }
 _EXTENSION_TO_MODULE.update({ext[1:]: ("imagefolder", {}) for ext in imagefolder.ImageFolder.IMAGE_EXTENSIONS})
+_EXTENSION_TO_MODULE.update({ext[1:].upper(): ("imagefolder", {}) for ext in imagefolder.ImageFolder.IMAGE_EXTENSIONS})

Original file line number	Diff line number	Diff line change
`@@ -43,3 +43,4 @@ def _hash_python_lines(lines: List[str]) -> str:`
`43`	`43`	`"txt": ("text", {}),`
`44`	`44`	`}`
`45`	`45`	`_EXTENSION_TO_MODULE.update({ext[1:]: ("imagefolder", {}) for ext in imagefolder.ImageFolder.IMAGE_EXTENSIONS})`
	`46`	`+_EXTENSION_TO_MODULE.update({ext[1:].upper(): ("imagefolder", {}) for ext in imagefolder.ImageFolder.IMAGE_EXTENSIONS})`