In order to avoid parsing exports with code snippets, let's give built-in functionality to transform a diffgram export object into a dataset object.

Something like this:

dataset = project.dataset.load_from_export('path/to/export/json')
dataset[0]
dataset.to_pytorch()