Is there any way of importing playwright's type hints in Python?

I'm literally doing the same that is shown in the sync example (https://playwright.dev/python/docs/api/class-playwright), passing the playwright context to a function that runs it, but inside my run function, playwright is Unknown, so chromium, browser, and everythin after is left as Any. Is there a way of solving this for better type hinting support? Thanks in advance!