From b344056932e3247d16ccb4f6a3af9e78cf0549e3 Mon Sep 17 00:00:00 2001
From: Maxime Beauchemin <maximebeauchemin@gmail.com>
Date: Thu, 21 Jun 2018 18:09:08 -0700
Subject: [PATCH] [bugfix] add support for numeric nodes in Sankey (#5154)

* [bugfix] add support for numeric nodes in Sankey

closes https://github.com/apache/incubator-superset/issues/5142

* lint
---
 superset/viz.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/superset/viz.py b/superset/viz.py
index cd1170984e..f3c7b154ea 100644
--- a/superset/viz.py
+++ b/superset/viz.py
@@ -31,6 +31,7 @@ from markdown import markdown
 import numpy as np
 import pandas as pd
 from pandas.tseries.frequencies import to_offset
+from past.builtins import basestring
 import polyline
 import simplejson as json
 from six import string_types, text_type
@@ -1596,6 +1597,8 @@ class SankeyViz(BaseViz):
 
     def get_data(self, df):
         df.columns = ['source', 'target', 'value']
+        df['source'] = df['source'].astype(basestring)
+        df['target'] = df['target'].astype(basestring)
         recs = df.to_dict(orient='records')
 
         hierarchy = defaultdict(set)