Fix package regression in PyTorch 2.7.1 training images #5470
bhanutejagk:master was force-pushed and no longer has any new commits.
Pushing new commits will allow the pull request to be re-opened.
Pushing new commits will allow the pull request to be re-opened.