Austin Schuh | 9049e20 | 2022-02-20 17:34:16 -0800 | [diff] [blame^] | 1 | import numpy as np |
2 | from scipy import sparse | ||||
3 | import utils.codegen_utils as cu | ||||
4 | |||||
5 | P = sparse.triu([[2., 5.], [5., 1.]], format='csc') | ||||
6 | q = np.array([3., 4.]) | ||||
7 | |||||
8 | A = sparse.csc_matrix([[-1., 0.], [0., -1.], [-1., 3.], [2., 5.], [3., 4]]) | ||||
9 | l = -np.inf * np.ones(A.shape[0]) | ||||
10 | u = np.array([0., 0., -15., 100., 80.]) | ||||
11 | |||||
12 | sols_data = {'sigma_new': 5} | ||||
13 | |||||
14 | # Generate problem data | ||||
15 | cu.generate_problem_data(P, q, A, l, u, 'non_cvx', sols_data) |