module 'vmoe.checkpoints.partitioned' has no attribute 'Mesh'

[nathanlmz]

Hi, I am facing this issue when running the example notebook (vmoe/vmoe/notebooks/demo_eee_CIFAR100.ipynb).
I haven't changed any code.