The dataset used in the paper is the Atari 2600 dataset, which consists of 49 games. The dataset is used to test the Successor Uncertainties algorithm.
Not to scale since I basically eyeballed it, but I think I got it close enough. I'm still learning modeling so I'm passing the savings on to you. :D Source: Objaverse 1.0 / Sketchfab