Edge_stats.json missing

I am trying to run “examples/cfd/vortex_shedding_mgn/inference_analysis/inference_analysis.ipynb” notebook file with Nvidia Modulus docker 24.04.
I downloaded the “cylinder_flow” dataset (meta.json, test.tfrecord, train.tfrecord, valid.tfrecord) using download_dataset.sh. However, when I run the notebook file, I am getting the error at the beginning:
FileNotFoundError: [Errno 2] No such file or directory: ‘edge_stats.json’
This is related to building the GNN using PyTorch, so it is not available at deepmind repo. Does anyone have an idea where to find this file?