def gen_dataset_func(do_shuffle: Optional[bool] = False) -> Dataset: test_datasource = TestDatasource(do_shuffle) return read_datasource(test_datasource)
def gen_dataset_func() -> Dataset: test_datasource = TestDatasource() return read_datasource(test_datasource)