changed function arguments

cvxgrp · Jan 11, 2022 · 2d5024a · 2d5024a
1 parent 1238d22
commit 2d5024a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cvxpylayers/torch/test_cvxpylayer.py b/cvxpylayers/torch/test_cvxpylayer.py
@@ -88,8 +88,8 @@ def test_least_squares(self):
         def lstsq(
             A,
             b): return torch.solve(
-            (A_th.t() @ b_th).unsqueeze(1),
-            A_th.t() @ A_th +
+            (A.t() @ b).unsqueeze(1),
+            A.t() @ A +
             torch.eye(n).double())[0]
         x_lstsq = lstsq(A_th, b_th)