[DataPipe] Fix missing imports in DataPipe interface file (#97458)
Fixes https://github.com/pytorch/data/issues/1106
Ran linter locally on `datapipes.pyi` (which is generated during installation) to confirm
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97458
Approved by: https://github.com/mikaylagawarecki