Fixing error with sqllab numpy array

(cherry picked from commit cf8a72b)
michellethomas · Mar 19, 2018 · d5cbcbf · d5cbcbf
1 parent 55e20ae
commit d5cbcbf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/superset/sql_lab.py b/superset/sql_lab.py
@@ -97,7 +97,7 @@ def convert_results_to_df(cursor_description, data):
     if data:
         first_row = data[0]
         has_dict_col = any([isinstance(c, dict) for c in first_row])
-        df_data = list(data) if has_dict_col else np.array(data)
+        df_data = list(data) if has_dict_col else np.array(data, dtype=object)
     else:
         df_data = []
 

diff --git a/tests/sqllab_tests.py b/tests/sqllab_tests.py
@@ -203,8 +203,8 @@ def test_alias_duplicate(self):
             raise_on_error=True)
 
     def test_df_conversion_no_dict(self):
-        cols = [['string_col'], ['int_col']]
-        data = [['a', 4]]
+        cols = [['string_col'], ['int_col'], ['float_col']]
+        data = [['a', 4, 4.0]]
         cdf = convert_results_to_df(cols, data)
 
         self.assertEquals(len(data), cdf.size)