What worked better for me was to go to Project Structure > Project Settings > Modules then select my module, on the right-hand side pane choose Dependencies, and from there, I set the Python SDK via the Module SDK: drop down
Project Structure > Project Settings > Modules
Dependencies
Module SDK: