import os
import sys

# Check if the system platform is Linux
if sys.platform.startswith('linux'):
    os.environ['PYOPENGL_PLATFORM'] = 'glx'

If you add these lines to the code it will use OpenGL for X11 instead of the EGL of wayland.