remove_dp

torchdata.dataloader2.graph.remove_dp(graph: Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], DataPipeGraph]]]], datapipe: Union[IterDataPipe, MapDataPipe]) → Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], Dict[int, Tuple[Union[IterDataPipe, MapDataPipe], DataPipeGraph]]]]

Given the graph of DataPipe generated by traverse_dps function and the DataPipe to be removed, return the new graph of DataPipe.

Note

• This function can not remove DataPipe that takes multiple DataPipes as the input.