Audit the LSSTDataset classes and make sure that the values that they are returning are numpy not torch tensors.
Audit the LSSTDataset classes and make sure that the values that they are returning are numpy not torch tensors.