@override Object toJson() => { if (trainingDatasetStats != null) 'trainingDatasetStats': trainingDatasetStats!.toJson(), };