def fake_dataset_fn(unused):
    del unused

    def make_dataset(time_us, num_elements):
        return dataset_ops.Dataset.range(num_elements).apply(sleep.sleep(time_us))
    return make_dataset((1000 * 1000), 0).concatenate(make_dataset(1000, 100)).take(100)
def fake_dataset_fn(unused):
    del unused

    def make_dataset(time_us, num_elements):
        return dataset_ops.Dataset.range(num_elements).apply(sleep.sleep(time_us))
    return make_dataset((1000 * 1000), 0).concatenate(make_dataset(1000, 100)).take(100)