torch 2.0.0 doesn't pass with mypy >= 1.9.0

Summary

https://github.com/pytorch/pytorch/issues/125320

https://github.com/python/mypy/issues/17189

Explain your context

(Explain what you did.)

What is the expected correct behavior? What was your goal?

(What you should see instead.)

What is the current bug behavior?

(What actually happens.)

Was there an error message?

(Paste any relevant logs - please use

code blocks like this one

or

this one

to format console output, logs, and code, as it's very hard to read otherwise.)

Steps to reproduce

(How one can reproduce the issue - this is very important)