Al final lo que he hecho ha sido descargarme el wake lock y lo he configurado de la siguiente manera.
en Adquire relase lock partial _wake - lock (cpu on, screen off, keyboard off) tambien activo show notification y start on boot le doy a la pestaña Adquire relase, y atrás para salirme de la pantalla.
Ahora cada vez que lo pongo en suspender sigue manteniendo mi wifi encendido con lo que sólo puedo escuchar las notificaciones.
Algo es algo
Espero haber ayudado a alguien que estuviera buscando una solución para dejar activo el wifi.
Tambien he probado las otras configuraciones, a mi sólo me funciona la que he mencionado anteriormente.