[fbsync] Add Path type annotation for ImageFolder (#8526)

NicolasHug · facebook-github-bot · commit c07f62772f00 · 2024-08-02T05:42:30.000-07:00
Reviewed By: vmoens

Differential Revision: D60596246

fbshipit-source-id: 956dfd072507c3205d0d928ef5d7c029b429505f
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,
