Hi @jennaliu thanks for your contribution!

Can you provide any kind of example in which the previous version fails but it works with the newer code?
Even better if you can add it to our test suite! =)

When opening a file with class or function definitions at the beginning of the file, Vim freezes. To test it, create a file with the following content (no empty lines at the beginning of the file), and open it with Vim. Sorry, I am not very familiar with the test suite right now. I submitted this pull request because it fixed my issue and I thought it could also help others.

class Class1:

    def __init__(self):
        super().__init__()


class Class2:

    def __init__(self):
        super().__init__()

Don't worry @jennaliu !!
Thanks for your contribution!

Thank you and thanks @diraol for your steady collaboration!