Add Path type annotation for ImageFolder

pytorch · Jul 15, 2024 · 46a64d9 · 46a64d9
1 parent 87d05c6
commit 46a64d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/datasets/folder.py b/torchvision/datasets/folder.py
@@ -318,7 +318,7 @@ class ImageFolder(DatasetFolder):
 
     def __init__(
         self,
-        root: str,
+        root: Union[str, Path],
         transform: Optional[Callable] = None,
         target_transform: Optional[Callable] = None,
         loader: Callable[[str], Any] = default_loader,