diff --git a/torchvision/prototype/datasets/__init__.py b/torchvision/prototype/datasets/__init__.py index b08ddb93c31..1945b5a5d9e 100644 --- a/torchvision/prototype/datasets/__init__.py +++ b/torchvision/prototype/datasets/__init__.py @@ -11,5 +11,5 @@ from ._home import home # Load this last, since some parts depend on the above being loaded first -from ._api import register, _list as list, info, load # usort: skip +from ._api import register, _list as list, info, load, find # usort: skip from ._folder import from_data_folder, from_image_folder