Fix bazel workflows (#70137)
Summary:
Fixes regression after manual rebase of https://github.com/pytorch/pytorch/commit/e35bf564611ac00886f6d745b0406d759e054fef
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70137
Reviewed By: pbelevich
Differential Revision: D33197055
Pulled By: malfet
fbshipit-source-id: 21adf7297f75715a59d2a1b3751b4ec8f71c7c03