def METHOD_NAME(
    data_source: Iterable[DataItem], transform_func: Callable[[DataItem], ModelInput], indices: List[int]
):
    for index in indices:
        yield transform_func(data_source[index])