While this is no problem in itself, this created an Error when using MultiVariateStudentT distribution in a higher level package, because it required batch_shape to be of type torch.Size

I don't see any issue with assigning the variable to a torch.Size object. Did I miss something?

batch_shape = torch.Size(broadcast_shape(df.shape, loc.shape[:-1], scale_tril.shape[:-2]))