Fix syntax issue
File didn't save I guess?
@@ -54,8 +54,8 @@ class MatmulLazyTensor(LazyTensor):
left_grad = self.left_lazy_tensor._quad_form_derivative(left_vecs, right_vecs_times_right_lazy_tensor)
right_grad = self.right_lazy_tensor._quad_form_derivative(left_vecs_times_left_lazy_tensor_t, right_vecs)
- left_grad = (left_grad,) if not isinstance(right_grad, tuple)
- right_grad = (right_grad,) if not isinstance(right_grad, tuple)
+ left_grad = (left_grad,) if not isinstance(left_grad, tuple) else left_grad
+ right_grad = (right_grad,) if not isinstance(right_grad, tuple) else right_grad
return left_grad + right_grad
def _size(self):
