Fix type hints of the callable arguments for DataLoader (#52924)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/52806
Pull Request resolved: https://github.com/pytorch/pytorch/pull/52924
Reviewed By: malfet
Differential Revision: D26694894
Pulled By: ejguan
fbshipit-source-id: 55734ec9684caa90f1e599b65659b7c57047f802
