I didn't realise I had to remove the build directory. Now it imports correctly.
For anyone that needs to know you need: extra_link_args=['-framework', 'OpenGL'] Delete the build directory and try it again. It will work.
extra_link_args=['-framework', 'OpenGL']