BUG: handle positional-only parameters in @ normalize

Author: ev-br

torch_np/_funcs.py

@@ -59,16 +59,18 @@ def normalizer(func):
     def wrapped(*args, **kwds):
         sig = inspect.signature(func)
 
-        dct = {}
+        lst, dct = [], {}
         # loop over positional parameters and actual arguments
         for arg, (name, parm) in zip(args, sig.parameters.items()):
             print(arg, name, parm.annotation)
             normalizer = normalizers.get(parm.annotation, None)
             if normalizer:
-                dct[name] = normalizer(arg, name)
+                # dct[name] = normalizer(arg, name)
+                lst.append(normalizer(arg))
             else:
                 # untyped arguments pass through
-                dct[name] = arg
+                # dct[name] = arg
+                lst.append(arg)
 
         # normalize keyword arguments
         for name, arg in kwds.items():
@@ -86,7 +88,7 @@ def wrapped(*args, **kwds):
             else:
                 dct[name] = arg
 
-        ba = sig.bind(**dct)
+        ba = sig.bind(*lst, **dct)
         ba.apply_defaults()
 
         # Now that all parameters have been consumed, check:

torch_np/tests/numpy_tests/lib/test_shape_base_.py

@@ -597,7 +597,7 @@ def test_basic(self):
         assert type(res) is np.ndarray
 
         aa = np.ones((3, 1, 4, 1, 1))
-        assert aa.squeeze().base is aa
+        assert aa.squeeze().get()._base is aa.get()
 
     def test_squeeze_axis(self):
         A = [[[1, 1, 1], [2, 2, 2], [3, 3, 3]]]