MAINT: move ufunc.__name__ and __qualname__ to the decorator

Author: ev-br

torch_np/_ufuncs.py

@@ -47,6 +47,9 @@ def wrapped(
         result = torch_func(*tensors)
         return result
 
+    wrapped.__qualname__ = torch_func.__name__
+    wrapped.__name__ = torch_func.__name__
+
     return wrapped
 
 
@@ -81,6 +84,10 @@ def matmul(
     return result
 
 
+matmul.__qualname__ = "matmul"
+matmul.__name__ = "matmul"
+
+
 def divmod(
     x1: ArrayLike,
     x2: ArrayLike,
@@ -116,16 +123,17 @@ def divmod(
     return quot, rem
 
 
+divmod.__qualname__ = "divmod"
+divmod.__name__ = "divmod"
+
+
 #
 # For each torch ufunc implementation, decorate and attach the decorated name
 # to this module. Its contents is then exported to the public namespace in __init__.py
 #
 for name in _binary:
     ufunc = getattr(_binary_ufuncs_impl, name)
     decorated = normalizer(deco_binary_ufunc(ufunc))
-
-    decorated.__qualname__ = name  # XXX: is this really correct?
-    decorated.__name__ = name
     vars()[name] = decorated
 
 
@@ -134,6 +142,10 @@ def modf(x, /, *args, **kwds):
     return rem, quot
 
 
+modf.__qualname__ = "modf"
+modf.__name__ = "modf"
+
+
 _binary = _binary + ["divmod", "modf", "matmul"]
 
 
@@ -178,6 +190,9 @@ def wrapped(
         result = torch_func(*tensors)
         return result
 
+    wrapped.__qualname__ = torch_func.__name__
+    wrapped.__name__ = torch_func.__name__
+
     return wrapped
 
 
@@ -188,9 +203,6 @@ def wrapped(
 for name in _unary:
     ufunc = getattr(_unary_ufuncs_impl, name)
     decorated = normalizer(deco_unary_ufunc(ufunc))
-
-    decorated.__qualname__ = name  # XXX: is this really correct?
-    decorated.__name__ = name
     vars()[name] = decorated