OGBN-LSMAG partitioning script failing

Hello everyone, I have spent quite some time trying to follow the official documentation of DGL for the partitioning of data of OGBN-LSMAG(https://www.dgl.ai/dgl_docs/guide/distributed-preprocessing.html#guide-distributed-preprocessing). However I am enountering some issues with the partitioning process.
We are using a metadata.json configuration file generated by chunk_graph.py However it doesn't contain many fields used by other script for dispatching like num_nodes_per_type causing the scripts to fail. Is it normal or am I missing something because the same file is the same one generated in the official documentation as well as Readme file