mlab through mayavi renders black screen

Im creating a project using the sfepy package, which piggybacks off mayavi for visualisation.

However I have no render!

Im running on ubuntu 20.04, intel i7-7700HQ and GTX1060 using the 460 (tested) driver.

All packages are seemingly installed correctly, I can even save screenshots of what im supposed to be looking at!

I copied the following example:
from numpy import pi, sin, cos, mgrid

dphi, dtheta = pi / 250.0, pi / 250.0
[phi, theta] = mgrid[0:pi + dphi * 1.5:dphi, 0:2 * pi + dtheta * 1.5:dtheta]
m0 = 4;
m1 = 3;
m2 = 2;
m3 = 3;
m4 = 6;
m5 = 2;
m6 = 6;
m7 = 4;
r = sin(m0 * phi) ** m1 + cos(m2 * phi) ** m3 + sin(m4 * theta) ** m5 + cos(m6 * theta) ** m7
x = r * sin(phi) * cos(theta)
y = r * cos(phi)
z = r * sin(phi) * sin(theta)

# View it.
from mayavi import mlab
s = mlab.mesh(x, y, z)
mlab.show()

Screenshot of output: (https://user-images.githubusercontent.com/86074725/122449760-21930e00-cf9e-11eb-9a1d-67ddddcd163f.png)

Any help would be gratefully received!