better fix

sigma-py · Jun 21, 2024 · 4799289 · 4799289
1 parent 7f756db
commit 4799289
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "npx"
-version = "0.1.3"
+version = "0.1.4"
 authors = [{name = "Nico Schlömer", email = "nico.schloemer@gmail.com"}]
 description = "Some useful extensions for NumPy"
 readme = "README.md"

diff --git a/src/npx/_unique.py b/src/npx/_unique.py
@@ -63,7 +63,7 @@ def unique_rows(a: ArrayLike, **kwargs) -> np.ndarray | tuple[np.ndarray, ...]:
     if isinstance(out, tuple):
         out = (out[0].view(a.dtype).reshape(out[0].shape[0], *a_shape[1:]), *out[1:])
         if len(out) > 1:
-            out = (out[0], out[1].T, *out[2:])
+            out = (out[0], out[1].flatten(), *out[2:])
     else:
         out = out.view(a.dtype).reshape(out.shape[0], *a_shape[1:])