MAINT: rm dead code

Author: ev-br

torch_np/_detail/_util.py

@@ -170,34 +170,6 @@ def typecast_tensors(tensors, target_dtype, casting):
     return tuple(cast_tensors)
 
 
-def axis_expand_func(func, tensor, axis, *args, **kwds):
-    """Generically handle axis arguments in reductions."""
-    if axis is not None:
-        if not isinstance(axis, (list, tuple)):
-            axis = (axis,)
-        axis = normalize_axis_tuple(axis, tensor.ndim)
-
-    if axis == ():
-        newshape = expand_shape(tensor.shape, axis=0)
-        tensor = tensor.reshape(newshape)
-        axis = (0,)
-
-    result = func(tensor, axis=axis, *args, **kwds)
-
-    return result
-
-
-def axis_ravel_func(func, tensor, axis, *args, **kwds):
-    """Generically handle axis arguments in cumsum/cumprod."""
-    if axis is not None:
-        axis = normalize_axis_index(axis, tensor.ndim)
-
-    tensors, axis = axis_none_ravel(tensor, axis=axis)
-    tensor = tensors[0]
-
-    result = func(tensor, axis=axis, *args, **kwds)
-
-    return result
 
 
 def _coerce_to_tensor(obj, dtype=None, copy=False, ndmin=0):

torch_np/_funcs.py

@@ -2,9 +2,8 @@
 
 import torch
 
-from . import _detail as _impl
+from ._detail import _dtypes_impl, _util, _reductions as _impl
 from . import _helpers
-from ._detail import _util
 from ._normalizations import (
     ArrayLike,
     AxisLike,
@@ -15,7 +14,6 @@
     normalizer,
 )
 
-_dtypes_impl = _impl._dtypes_impl
 
 NoValue = _util.NoValue
 

torch_np/_wrapper.py

@@ -8,8 +8,7 @@
 
 import torch
 
-from . import _detail as _impl
-from . import _dtypes, _funcs, _helpers
+from . import _dtypes
 from ._detail import _dtypes_impl, _util
 from ._ndarray import asarray
 from ._normalizations import (
@@ -1036,7 +1035,7 @@ def put_along_axis(arr: ArrayLike, indices: ArrayLike, values: ArrayLike, axis):
     return None
 
 
-# ### unqiue et al ###
+# ### unique et al ###
 
 
 @normalizer