ValueError: Optional arg `num_classes` must be type `int` when task is multiclass. Got <class 'NoneType'>