fix: Crash caused by numpy.vectorize

Update result_set.py Update result_set.py
apache · Oct 25, 2022 · 83d8739 · 83d8739
1 parent 1388f21
commit 83d8739
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/superset/result_set.py b/superset/result_set.py
@@ -63,8 +63,13 @@ def stringify(obj: Any) -> str:
 
 
 def stringify_values(array: np.ndarray) -> np.ndarray:
-    vstringify = np.vectorize(stringify)
-    return vstringify(array)
+    result = np.copy(array)
+
+    with np.nditer(result, flags=["refs_ok"], op_flags=["readwrite"]) as it:
+        for obj in it:
+            obj[...] = stringify(obj)
+
+    return result
 
 
 def destringify(obj: str) -> Any: