Add Problem.load and expose Problem

CHANGELOG.md

@@ -4,6 +4,11 @@ All notable changes to this project will be documented in this file.
 
 ## [Unreleased]
 
+### Added
+
+* Expose `Problem` from top-level module
+* `Problem.load` utility function
+
 ## [2.0.0] - 2023-12-11
 
 ### Added

qpbenchmark/__init__.py

@@ -17,6 +17,7 @@
 
 """Benchmark for quadratic programming solvers available in Python."""
 
+from .problem import Problem
 from .report import Report
 from .results import Results
 from .run import run
@@ -27,6 +28,7 @@
 __version__ = get_version()
 
 __all__ = [
+    "Problem",
     "Report",
     "Results",
     "TestSet",

qpbenchmark/problem.py

@@ -85,3 +85,24 @@ def to_sparse(self):
             self.ub,
             name=self.name,
         )
+
+    @staticmethod
+    def load(file: str, name: str):
+        """Load problem from file.
+
+        Args:
+            file: Path to the file to read.
+            name: Problem name.
+        """
+        loaded = qpsolvers.Problem.load(file)
+        return Problem(
+            loaded.P,
+            loaded.q,
+            loaded.G,
+            loaded.h,
+            loaded.A,
+            loaded.b,
+            loaded.lb,
+            loaded.ub,
+            name,
+        )