Fix type warning (#46770)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46770
Test Plan: Standard pre-commit test rig.
Reviewed By: malfet
Differential Revision: D24480898
fbshipit-source-id: a5031f1e20f4b1ea5954e7cabd54502300d5a916
