Thanks for the example.
An easy way of “named samples” would be to return a dict in your Dataset and use the key as the name.
Not sure, if a custom class implementation would work, and I would need to verify it.