diffapqp 0.1.0__py3-none-any.whl

lao/diffapqp/solve_result.py

@@ -0,0 +1,55 @@
+"""Small data containers for the batched CVXPY solve pipeline."""
+
+from dataclasses import dataclass
+
+import cvxpy as cp
+import numpy as np
+
+
+@dataclass
+class SampleSolveResult:
+    """One sample's standardized QP solve result."""
+
+    primal: np.ndarray
+    dual_eq: np.ndarray
+    dual_ineq: np.ndarray
+    value: float
+    raw_solution: object
+    timing: dict
+
+
+@dataclass
+class PreparedSolveData:
+    """CVXPY data prepared for one sample before entering the solver."""
+
+    cvxpy_prob: cp.Problem
+    data: dict
+    chain: object
+    inverse_data: object
+    prepare_time: float
+
+
+@dataclass
+class PreparedSolveTask:
+    """One solve task with prepared CVXPY data and solver options."""
+
+    prepared: PreparedSolveData
+    solver: str
+    warm_start: bool
+    verbose: bool
+    solver_args: dict
+    n_eq_standard: int
+    n_ineq_standard: int
+
+
+@dataclass
+class LayerSolveSummary:
+    """Batched solve summary needed by the solution-map layer."""
+
+    primal: np.ndarray
+    dual_eq: np.ndarray
+    dual_ineq: np.ndarray
+    raw_solutions: list
+    timing: dict
+    active_ineq_indices: list
+    inequality_rhs: np.ndarray