class_to_idx parameter should not be a required parameter for regression tasks

[Basel Mousi]

Hello,

I was trying to get the top neurons associated with a linear regression probe on a set of synthetic data. I obtained that the class_to_idx is a required positional argument; Shouldn't it be optional for regression tasks?

Thanks in advance