<filename>pretorched/data/__init__.py
from .datasets.image import ImageDataset, ImageFolder
# from .datasets.video import VideoRecordDataset, RecordSet
from .datasets import VideoRecordDataset, RecordSet
from . import utils

__all__ = [
    'ImageDataset',
    'ImageFolder',
    'VideoRecordDataset',
    'RecordSet',
    'utils'
]

<filename>pretorched/data/__init__.py
from .datasets.image import ImageDataset, ImageFolder
# from .datasets.video import VideoRecordDataset, RecordSet
from .datasets import VideoRecordDataset, RecordSet
from . import utils

__all__ = [
    'ImageDataset',
    'ImageFolder',
    'VideoRecordDataset',
    'RecordSet',
    'utils'
]

# from .datasets.video import VideoRecordDataset, RecordSet