Refactor evaluate_and_store method in Leva module to store experime…

…nt and runner result
kieranklaassen · Aug 23, 2024 · 7f44b59 · 7f44b59
1 parent 44ad19c
commit 7f44b59
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/leva.rb b/lib/leva.rb
@@ -82,6 +82,10 @@ def evaluate(prediction, recordable)
     # @param runner_result [Leva::RunnerResult] The runner result to evaluate.
     # @return [Leva::EvaluationResult] The stored evaluation result.
     def evaluate_and_store(experiment, runner_result)
+      @experiment = experiment
+      @runner_result = runner_result
+
+      # TODO: return parsed predfiction maybe?
       score = evaluate(runner_result.prediction, runner_result.dataset_record.recordable)
 
       EvaluationResult.create!(