[DataPipe] fixing tests related fork() to remove warnings (#64827)
authorKevin Tse <ktse@fb.com>
Fri, 10 Sep 2021 18:00:01 +0000 (11:00 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Fri, 10 Sep 2021 18:01:42 +0000 (11:01 -0700)
commit5060b69d62943e11f053f8c86b8154a7d009af11
tree58b394786e9a2f041b4d732f02b8490561f4fa49
parentade4bf3e82ea369e77d511a796ba936ce281aa89
[DataPipe] fixing tests related fork() to remove warnings (#64827)

Summary:
There are two warnings produced by `test_fork_datapipe`. This PR addresses the issues raised by those warnings without impacting the test cases.

cc VitalyFedyunin ejguan

Pull Request resolved: https://github.com/pytorch/pytorch/pull/64827

Reviewed By: ejguan

Differential Revision: D30870528

Pulled By: NivekT

fbshipit-source-id: 580a001c6fa3ff6f8b04a7e5183e58861938204b
test/test_datapipe.py