[MPS] Fix the regression with test_index_select_scalar() (#94645)
The PR #94347 caused a regression in test_mps which this patch fixes it.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94645
Approved by: https://github.com/DenisVieriu97