Error when running with airfoil dataset

When I load the airfoil dataset and run the code, I get an error

```
Cannot bitcast from 3 to 9 [Op:Bitcast]
```
Right around this line number:
https://github.com/deepmind/deepmind-research/blob/11c2ab53e8afd24afa8904f22fd81b699bfbce6e/meshgraphnets/common.py#L44

Do you see the same error when you run?
I am trying to write a simple matplotlib to plot the mesh. This would help me to understand how the edges are connected to each other. Description on edge connectivity for tensor flow graph networks is a bit lacking.