From 1df4fa26c25e63330ec7d8bfc93f1f13a418705f Mon Sep 17 00:00:00 2001
From: Ville Brofeldt <33317356+villebro@users.noreply.github.com>
Date: Mon, 24 Jun 2019 01:53:26 +0300
Subject: [PATCH] Fix deck.gl sample charts with js (#7749)

* Fix js_data_mutator

* Remove redundant line change

* Add missing line changes
---
 superset/data/deck.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/superset/data/deck.py b/superset/data/deck.py
index 80e75810a4..398255cce0 100644
--- a/superset/data/deck.py
+++ b/superset/data/deck.py
@@ -416,10 +416,11 @@ def load_deck_dash():
             'population',
             'area',
         ],
-        'js_datapoint_mutator':
-            '(d) => {\n    d.elevation = d.extraProps.population/d.extraProps.area/10\n \
-         d.fillColor = [d.extraProps.population/d.extraProps.area/60,140,0]\n \
-         return d;\n}',
+        'js_data_mutator':
+            'data => data.map(d => ({\n'
+            '    ...d,\n'
+            '    elevation: d.extraProps.population/d.extraProps.area/10,\n'
+            '}));',
         'js_tooltip': '',
         'js_onclick_href': '',
         'where': '',
@@ -533,8 +534,10 @@ def load_deck_dash():
         'js_columns': [
             'color',
         ],
-        'js_datapoint_mutator': 'd => {\n    return {\n        ...d,\n        color: \
-            colors.hexToRGB(d.extraProps.color),\n    }\n}',
+        'js_data_mutator': 'data => data.map(d => ({\n'
+                           '    ...d,\n'
+                           '    color: colors.hexToRGB(d.extraProps.color)\n'
+                           '}));',
         'js_tooltip': '',
         'js_onclick_href': '',
         'where': '',