They implemented their own getitem dunder (which grabs the var[stuff]) but... They don't use the index (the thing inside the []) at all in it Basically no matter what you put in the [], you get an unrelated sample
Can you make a recommendation how to implement it? They obviously don't need the key because I think they just generate a sample "on the go". So what does one do then?