Undo hack to test dask shape

Author: crusaderky

Diff for: src/array_api_extra/_lib/_funcs.py

@@ -552,7 +552,7 @@ def nunique(x: Array, /, *, xp: ModuleType | None = None) -> Array:
     _, counts = xp.unique_counts(x)
     n = _compat.size(counts)
     # FIXME https://github.com/data-apis/array-api-compat/pull/231
-    if n is None or math.isnan(n):  # e.g. Dask, ndonnx
+    if n is None:  # e.g. Dask, ndonnx
         return xp.astype(counts, xp.bool).sum()
     return xp.asarray(n, device=_compat.device(x))