diff --git a/trainer/lib/trainer/test_parser.rb b/trainer/lib/trainer/test_parser.rb index c973043d290..933ecc26c67 100644 --- a/trainer/lib/trainer/test_parser.rb +++ b/trainer/lib/trainer/test_parser.rb @@ -337,7 +337,16 @@ def summaries_to_data(summaries, failures, output_remove_retry_attempts: false) row end - self.data = rows + self.data = remove_repetition_retries(rows, output_remove_retry_attempts) + end + + def remove_repetition_retries(rows, output_remove_retry_attempts) + if output_remove_retry_attempts + rows.group_by { |row| row[:test_name] } + .map { |_, group| group.min_by { |row| row[:number_of_failures_excluding_retries] } } + else + rows + end end def test_summaries_to_configuration_names(test_summaries)