BUG: fix an issue where array functions would raise UnitConsistencyEr…

…ror on unyt arrays using non-default unit registries
yt-project · Nov 1, 2023 · 9dbd8ec · 9dbd8ec
1 parent 8754a5c
commit 9dbd8ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/unyt/_array_functions.py b/unyt/_array_functions.py
@@ -216,8 +216,8 @@ def _validate_units_consistency(objs):
     # because it's already a necessary condition for numpy to use our
     # custom implementations
     units = get_units(objs)
-    sunits = set(units)
-    if len(sunits) == 1:
+    unique_units = set(units)
+    if len(unique_units) == 1 or all(u.is_dimensionless for u in unique_units):
         return units[0]
     else:
         raise UnitInconsistencyError(*units)