@override Object toJson() => { if (trainingDatasetStats case final trainingDatasetStats?) 'trainingDatasetStats': trainingDatasetStats.toJson(), };