Compatibility layer between stable datasets and prototype transforms

torchvision/prototype/features/__init__.py

@@ -1,6 +1,7 @@
 from ._bounding_box import BoundingBox, BoundingBoxFormat
 from ._encoded import EncodedData, EncodedImage, EncodedVideo
-from ._feature import _Feature, FillType, FillTypeJIT, InputType, InputTypeJIT, is_simple_tensor
+
+from ._feature import _Feature, FillType, FillTypeJIT, GenericFeature, InputType, InputTypeJIT, is_simple_tensor
 from ._image import (
     ColorSpace,
     Image,
@@ -13,3 +14,5 @@
 )
 from ._label import Label, OneHotLabel
 from ._mask import Mask
+
+from ._dataset_wrapper import VisionDatasetFeatureWrapper  # usort: skip