Should custom Dataset always return tensors?

[AlexisW]

Have you ever tried to return an integer itself? That will also be converted to a tensor…
The question then is whether we want to explicitly convert that.