Hi everyone!
I have changed the def getitem(self,idx) function based on the solution from this post and I have encountered another issue that I do not know how to solve. Any input on how to fix it is much appreciated!

image1341×420 60.7 KB