Thanks for diving into it 
The URL you shared seems to point to a dynamic code snippet (from the master branch), that got some changes a few days ago. For future reference, the “part that takes a lot of time” is here
Do you know by any chance what part of the code exactly in the Dataset initialization caused some slowdowns on your side ?