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