ray.data.DatasetPipeline.select_columns#

DatasetPipeline.select_columns(cols: List[str], *, compute: Optional[str] = None, **ray_remote_args) → ray.data.dataset_pipeline.DatasetPipeline[ray.data.block.U][source]#

Apply Dataset.select_columns to each dataset/window in this pipeline.