def METHOD_NAME(samples: List[Tuple[np.ndarray, int]], input_name: str):
    def transform_fn(data_item):
        inputs = data_item
        return {input_name: [inputs]}
    return Dataset(samples, transform_fn)