class Riffer::Evals::RunResult

# File lib/riffer/evals/run_result.rb, line 11
def initialize(scenario_results:)
  @scenario_results = scenario_results
end

# File lib/riffer/evals/run_result.rb, line 19
def scores
  return {} if scenario_results.empty?

  totals = Hash.new(0.0)
  counts = Hash.new(0)

  scenario_results.each do |scenario|
    scenario.scores.each do |evaluator, score|
      totals[evaluator] += score
      counts[evaluator] += 1
    end
  end

  averages = {} #: Hash[singleton(Riffer::Evals::Evaluator), Float]
  totals.each_with_object(averages) do |(evaluator, total), hash|
    hash[evaluator] = total / counts[evaluator]
  end
end

# File lib/riffer/evals/run_result.rb, line 42
def to_h
  {
    scores: scores.transform_keys(&:name),
    scenario_results: scenario_results.map(&:to_h)
  }
end