Small fix for pandas unique utility

holoviz · Feb 13, 2018 · 3c89e27 · 3c89e27
1 parent 383690d
commit 3c89e27
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/holoviews/core/util.py b/holoviews/core/util.py
@@ -838,7 +838,10 @@ def unique_array(arr):
     if not len(arr):
         return arr
     elif pd:
-        return pd.unique(arr)
+        unique = pd.unique(arr)
+        if isinstance(arr, np.ndarray) and unique.dtype != arr.dtype:
+            return unique.astype(arr.dtype)
+        return unique
     else:
         arr = np.asarray(arr)
         _, uniq_inds = np.unique(arr, return_index=True)