I have following similar directory structure.
main.py pack |___sub_pack1 |__a.py |___sub_pack2 |__b.py
Go to settings-> project structure and sub_pack2 as a source