lint

Author: ev-br

torch_np/_detail/_flips.py

@@ -5,6 +5,7 @@
 
 from . import _scalar_types, _util
 
+
 def flip(m_tensor, axis=None):
     # XXX: semantic difference: np.flip returns a view, torch.flip copies
     if axis is None:
@@ -39,7 +40,7 @@ def rollaxis(tensor, axis, start=0):
         start += n
     msg = "'%s' arg requires %d <= %s < %d, but %d was passed in"
     if not (0 <= start < n + 1):
-        raise _util.AxisError(msg % ('start', -n, 'start', n + 1, start))
+        raise _util.AxisError(msg % ("start", -n, "start", n + 1, start))
     if axis < start:
         # it's been removed
         start -= 1

torch_np/_ndarray.py

@@ -10,7 +10,7 @@
     dtype_to_torch,
     emulate_out_arg,
 )
-from ._detail import _reductions, _util, _flips
+from ._detail import _flips, _reductions, _util
 
 newaxis = None
 

torch_np/_wrapper.py

@@ -8,7 +8,7 @@
 import torch
 
 from . import _dtypes, _helpers
-from ._detail import _reductions, _util, _flips
+from ._detail import _flips, _reductions, _util
 from ._ndarray import (
     array,
     asarray,
@@ -476,15 +476,16 @@ def broadcast_arrays(*args, subok=False):
 
 @asarray_replacer()
 def moveaxis(a, source, destination):
-    source = _util.normalize_axis_tuple(source, a.ndim, 'source')
-    destination = _util.normalize_axis_tuple(destination, a.ndim, 'destination')
+    source = _util.normalize_axis_tuple(source, a.ndim, "source")
+    destination = _util.normalize_axis_tuple(destination, a.ndim, "destination")
     return asarray(torch.moveaxis(a, source, destination))
 
 
 def swapaxis(a, axis1, axis2):
     arr = asarray(a)
     return arr.swapaxes(axis1, axis2)
 
+
 @asarray_replacer()
 def rollaxis(a, axis, start=0):
     return _flips.rollaxis(a, axis, start)
@@ -665,8 +666,10 @@ def prod(
         axis=axis, dtype=dtype, out=out, keepdims=keepdims, initial=initial, where=where
     )
 
+
 product = prod
 
+
 def cumprod(a, axis=None, dtype=None, out=None):
     arr = asarray(a)
     return arr.cumprod(axis=axis, dtype=dtype, out=out)