Translate CVXPY problems into gurobipy models
Project description
CVXPY x GUROBI
This small library provides an alternative way to solve CVXPY problems with Gurobi.
Usage
The library provides a solver that will translate a CVXPY Problem
into a gurobipy.Model
,
and optimize using Gurobi:
import cvxpy as cp
import cvxpy_gurobi
problem = cp.Problem(cp.Maximize(cp.Variable(name="x", nonpos=True)))
cvxpy_gurobi.solve(problem)
assert problem.value == 0
The solver can also be registered with CVXPY and used as any other solver:
import cvxpy as cp
from cvxpy_gurobi import GUROBI_TRANSLATION, solver
cvxpy_gurobi.register_solver()
# ^ this is the same as:
cp.Problem.register_solve_method(GUROBI_TRANSLATION, solver())
problem.solve(method=GUROBI_TRANSLATION)
This solver is a simple wrapper for the most common use case:
from cvxpy_gurobi import build_model, backfill_problem
model = build_model(problem)
model.optimize()
backfill_problem(problem, model)
assert model.optVal == problem.value
The build_model
function provided by this library translates the cvxpy.Problem
instance
into an equivalent gurobipy.Model
, and backfill_problem
sets the optimal
values on the original problem.
[!NOTE] Both functions must be used together as they rely on naming conventions to map variables and constraints between CVXPY and Gurobi.
The output of the build_model
function is a standard gurobipy.Model
instance,
which can be further customized prior to solving. This approach enables you to
manage how the model will be optimized.
Installation
pip install cvxpy-gurobi
CVXPY has an interface to Gurobi, why is this needed?
When using CVXPY's interface to Gurobi,
the problems fed to Gurobi have been pre-compiled by CVXPY,
meaning the model is not exactly the same as the one you have written.
This is great for solvers with low-level APIs, such as SCS or OSQP,
but gurobipy
allows you to express your models at a higher-level.
Providing the raw model to Gurobi is a better idea in general since the Gurobi solver is able to compile the problem with a better accuracy. The chosen algorithm can also be different depending on the way it is modelled, potentially leading to better performance.
In addition, CVXPY does not give access to the model before solving it.
CVXPY must therefore make some choices for you,
such as setting QCPDual
to 1 on all non-MIP models.
Having access to the model can help
if you want to handle the call to .optimize()
in a non-standard way,
e.g. by sending it to an async loop.
Example
Consider this QP problem:
import cvxpy as cp
x = cp.Variable(name="x")
problem = cp.Problem(cp.Minimize((x-1) ** 2))
The problem will be sent to Gurobi as (in LP format):
Minimize
[ 2 C0 ^2 ] / 2
Subject To
R0: - C0 + C1 = 1
Bounds
C0 free
C1 free
End
Using this package, it will instead send:
Minimize
- 2 x + Constant + [ 2 x ^2 ] / 2
Subject To
Bounds
x free
Constant = 1
End
Note that:
- the variable's name matches the user-defined problem;
- no extra (free) variables;
- no extra constraints.
Why not use gurobipy
directly?
CVXPY has 2 main features: a modelling API and interfaces to many solvers. The modelling API has a great design, whereas gurobipy
feels like a thin layer over the C API. The interfaces to other solvers can be useful to not have to rewrite the problem when switching solvers.
Supported versions
All supported versions of Python, CVXPY and gurobipy
should work.
However, due to licensing restrictions, old versions of
gurobipy
cannot be tested in CI.
If you run into a bug, please open an issue in this repo specifying
the versions used.
Contributing
It is highly recommended to use Hatch for development. It will handle all virtual environment management.
To lint and format the code, run:
hatch fmt
For testing, run:
hatch test
This will test the latest version of dependencies. You can also run hatch test --all
to test several combinations of the supported version range.
Make sure any change is tested through a snapshot test. To add a new test case,
build a simple CVXPY problem in tests/test_problems.py
in the appropriate category,
then run:
hatch run update-snapshots
You can then check the output in the tests/snapshots
folder is as expected.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file cvxpy_gurobi-1.0.0.tar.gz
.
File metadata
- Download URL: cvxpy_gurobi-1.0.0.tar.gz
- Upload date:
- Size: 44.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | f15e8565c66902f474b85ea728a49d6da5b70c5740c9153aa4593a4be9b76d38 |
|
MD5 | ee9ce3d9b840331c654995b2de8e645e |
|
BLAKE2b-256 | 45d17f9227a608207cb2b937d18f620a945db18bf8b117ac2495145d743cf7e4 |
File details
Details for the file cvxpy_gurobi-1.0.0-py3-none-any.whl
.
File metadata
- Download URL: cvxpy_gurobi-1.0.0-py3-none-any.whl
- Upload date:
- Size: 15.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 676243598cac0623a583b3df296ed6372b906e3114bcec7fff51ae650ed0fb79 |
|
MD5 | 41dcbc44fa1e7d0ae026664691cd4a0b |
|
BLAKE2b-256 | 6c97abba86114076963e1f1c1956ddaa52c37355d458a95b7efd759620fde1e6 |