110: anticipate future numpy deprecation

nicodv · Mar 21, 2019 · f06831c · f06831c
1 parent 8b1c614
commit f06831c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kmodes/util/__init__.py b/kmodes/util/__init__.py
@@ -61,4 +61,4 @@ def decode_centroids(encoded, mapping):
 
 def get_unique_rows(a):
     """Gets the unique rows in a numpy array."""
-    return np.vstack({tuple(row) for row in a})
+    return np.vstack(list({tuple(row) for row in a}))