Merge pull request #74 from asmeurer/sign-fix

Fix the definition of sign() for complex numbers

Author: asmeurer

array_api_strict/_elementwise_functions.py

@@ -855,6 +855,8 @@ def sign(x: Array, /) -> Array:
     """
     if x.dtype not in _numeric_dtypes:
         raise TypeError("Only numeric dtypes are allowed in sign")
+    if x.dtype in _complex_floating_dtypes:
+        return x/abs(x)
     return Array._new(np.sign(x._array), device=x.device)