map

open override fun map(input: TrainingState): TrainingUiState