ruff

54769ac9 - [ty] While loop modeling cleanup (#18994)

[ty] While loop modeling cleanup (#18994) ## Summary I found the previous code here very confusing, and it also did some unnecessary work. Hopefully this is a bit easier to understand.