def dataset_fn():
    return dataset_ops.Dataset.range(1).repeat().apply(interleave_ops.parallel_interleave(_make_fake_dataset_fn(), cycle_length=10))
def dataset_fn():
    return dataset_ops.Dataset.range(1).repeat().apply(interleave_ops.parallel_interleave(_make_fake_dataset_fn(), cycle_length=10))