diff --git a/kmodes/util/__init__.py b/kmodes/util/__init__.py index ba57170..6111967 100644 --- 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}))