mypy

9012fc9e - Some cleanup in partial plugin (#17423)

Some cleanup in partial plugin (#17423) Fixes https://github.com/python/mypy/issues/17405 Apart from fixing the crash I fix two obvious bugs I noticed while making this PR.