Fix issue 38095 TODOs in test_jit (#82629)
Fix TODOs related to https://github.com/pytorch/pytorch/issues/38095 in test_jit.py
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82629
Approved by: https://github.com/clee2000, https://github.com/malfet