fixes for trainTest reporting wrong metrics values sometimes