From 7c109c575016eb0596e20180b334ca2955bd33e6 Mon Sep 17 00:00:00 2001
From: Tiago Botelho <tiagonbotelho@hotmail.com>
Date: Fri, 9 Feb 2018 14:10:58 +0000
Subject: Refactor CycleAnalytics base stage class and model

---
 app/serializers/analytics_stage_entity.rb | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'app/serializers')

diff --git a/app/serializers/analytics_stage_entity.rb b/app/serializers/analytics_stage_entity.rb
index 564612202b5..ed1103b2064 100644
--- a/app/serializers/analytics_stage_entity.rb
+++ b/app/serializers/analytics_stage_entity.rb
@@ -6,7 +6,13 @@ class AnalyticsStageEntity < Grape::Entity
   expose :legend
   expose :description
 
-  expose :median, as: :value do |stage|
-    stage.median && !stage.median.zero? ? distance_of_time_in_words(stage.median) : nil
+  expose :medians, as: :values do |stage|
+    medians = stage.medians
+
+    unless medians.blank?
+      medians.each do |id, median|
+        medians[id] = distance_of_time_in_words(median)
+      end
+    end
   end
 end
-- 
cgit v1.2.1