What to modify to use other dataset?

[Jatinkalal]

Hey, Iam new to this field of human motion simulation, i wanted to try with other dataset such as KIT-ML or Human ML 3D which has 24 joints, I assume the CMU MoCap has 29 joints, what changes I need to do in the code? Also if you could provide any documentation to use mujoco_py that would be helpful too.
Thank You.