I recommend to use black to format your code.
Just a personal preference: if the code in one line is getting too long, put the comment regarding the line on top of it, not in the same line to stay below a certain line length. I recommend 120, you can specify this in black.
Do not use * imports:
When I go ...