Re-enable the signature fix in get_xp

Author: asmeurer

array_api_compat/_internal.py

@@ -22,12 +22,11 @@ def func(x, /, xp, kwarg=None):
 
     """
     def inner(f):
-        sig = signature(f)
-
         @wraps(f)
         def wrapped_f(*args, **kwargs):
             return f(*args, xp=xp, **kwargs)
 
+        sig = signature(f)
         new_sig = sig.replace(parameters=[sig.parameters[i] for i in sig.parameters if i != 'xp'])
 
         if wrapped_f.__doc__ is None:
@@ -38,7 +37,7 @@ def wrapped_f(*args, **kwargs):
 specification for more details.
 
 """
-        # wrapped_f.__signature__ = new_sig
+        wrapped_f.__signature__ = new_sig
         return wrapped_f
 
     return inner