Denisfy the matrices for flattening in testing.

osqp · Jul 18, 2024 · 85e0c36 · 85e0c36
1 parent 29b87a3
commit 85e0c36
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/osqp/tests/nn_test.py b/src/osqp/tests/nn_test.py
@@ -133,7 +133,7 @@ def f(P):
 
         return 0.5 * np.sum(np.square(x_hat - true_x))
 
-    dP_fd = approx_fprime(P.tolil().reshape((1,n*n)), f, epsilon=EPS)
+    dP_fd = approx_fprime(P.toarray().flatten(), f, epsilon=EPS)
     if verbose:
         print('dP_fd: ', np.round(dP_fd, decimals=4))
         print('dP: ', np.round(dP, decimals=4))
@@ -164,7 +164,7 @@ def f(A):
 
         return 0.5 * np.sum(np.square(x_hat - true_x))
 
-    dA_fd = approx_fprime(A.tolil().reshape((1,m*n)), f, epsilon=EPS)
+    dA_fd = approx_fprime(A.toarray().flatten(), f, epsilon=EPS)
     if verbose:
         print('dA_fd: ', np.round(dA_fd, decimals=4))
         print('dA: ', np.round(dA, decimals=4))