correct linter

Author: pgmoka

torch_xla/distributed/spmd/xla_sharding.py

@@ -704,14 +704,12 @@ def _einsum_linear_backward(grad_output: Tensor, input: Tensor, weight: Tensor,
     grad_input = grad_weight = grad_bias = None
 
     if needs_input_grad_input:
-      grad_input = torch.einsum('...m,mn->...n',
-                                               (grad_output, weight))
+      grad_input = torch.einsum('...m,mn->...n', (grad_output, weight))
     else:
       grad_input = None
 
     if needs_input_grad_weight:
-      grad_weight = torch.einsum('...m,...n->mn',
-                                                (grad_output, input))
+      grad_weight = torch.einsum('...m,...n->mn', (grad_output, input))
     else:
       grad_weight = None