style: un_flatten_grad

Author: kozistr

pytorch_optimizer/optimizer/utils.py

@@ -48,7 +48,7 @@ def un_flatten_grad(grads: torch.Tensor, shapes: List[int]) -> List[torch.Tensor
     un_flatten_grads: List[torch.Tensor] = []
     for shape in shapes:
         length = np.prod(shape)
-        un_flatten_grads.append(grads[idx : idx + length].view(shape).clone())
+        un_flatten_grads.append(grads[idx:idx + length].view(shape).clone())  # fmt: skip
         idx += length
     return un_flatten_grads