Replace group_by_key by group_by IterDataPipe (#64220)
authorErjia Guan <erjia@fb.com>
Tue, 31 Aug 2021 01:41:08 +0000 (18:41 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 31 Aug 2021 01:45:44 +0000 (18:45 -0700)
commitaf85bc5ffd1d4ad52e0fed255aa7afe2fdfbc5e2
tree9cbe6d76bfa3f489f49b39e9a9d1a7f7991acac4
parent4bd03b02424d93b72f15e28c542ede13f88ea929
Replace group_by_key by group_by IterDataPipe (#64220)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64220

Remove `ByKeyGrouperIterDataPipe` due to duplicated functionality.
Fix a bug in `GrouperIterDataPipe` using the existing test.

Test Plan: Imported from OSS

Reviewed By: VitalyFedyunin

Differential Revision: D30650542

Pulled By: ejguan

fbshipit-source-id: 666b4d28282fb4f49f3ff101b8d08be16a50d836
test/test_datapipe.py
torch/utils/data/datapipes/iter/__init__.py
torch/utils/data/datapipes/iter/grouping.py