openscvx 0.4.1.dev167__tar.gz → 0.4.1.dev170__tar.gz

{openscvx-0.4.1.dev167/openscvx.egg-info → openscvx-0.4.1.dev170}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openscvx
-Version: 0.4.1.dev167
+Version: 0.4.1.dev170
 Summary: A general Python-based successive convexification implementation which uses a JAX backend.
 Author-email: Chris Hayner and Griffin Norris <haynec@uw.edu>
 License: Apache Software License

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170}/openscvx/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.4.1.dev167'
-__version_tuple__ = version_tuple = (0, 4, 1, 'dev167')
+__version__ = version = '0.4.1.dev170'
+__version_tuple__ = version_tuple = (0, 4, 1, 'dev170')
 
-__commit_id__ = commit_id = 'g6aeaaf49b'
+__commit_id__ = commit_id = 'ga135b26e1'

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170}/openscvx/algorithms/__init__.py

@@ -129,6 +129,7 @@ class PenalizedTrustRegionConfig(BaseModel):
         return v
 
     k_max: int = 200
+    t_max: Optional[float] = None
     lam_prox: Union[float, Dict[str, Any]] = 1e-1
     lam_vc: Union[float, Dict[str, Any]] = 1e0
     lam_cost: Union[float, Dict[str, Any]] = 1e-2

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170}/openscvx/algorithms/base.py

@@ -718,6 +718,10 @@ class Algorithm(ABC):
     #: Maximum number of SCP iterations. Subclasses must set this in ``__init__``.
     k_max: int
 
+    #: Optional wall-clock time limit in seconds. ``None`` means no limit.
+    #: Subclasses must set this in ``__init__``.
+    t_max: Optional[float]
+
     @abstractmethod
     def initialize(
         self,

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170}/openscvx/algorithms/penalized_trust_region.py

@@ -87,6 +87,7 @@ class PenalizedTrustRegion(Algorithm):
         self,
         autotuner: "AutotuningBase" = None,
         k_max: int = 200,
+        t_max: float | None = None,
         lam_prox: Union[float, Dict[str, Union[float, list]]] = 1e-1,
         lam_vc: Union[float, Dict[str, Union[float, list]]] = 1e0,
         lam_cost: Union[float, Dict[str, float]] = 1e-2,
@@ -103,6 +104,8 @@ class PenalizedTrustRegion(Algorithm):
             autotuner: Weight adaptation strategy. Defaults to
                 :class:`AugmentedLagrangian` when ``None``.
             k_max: Maximum SCP iterations. Defaults to 200.
+            t_max: Wall-clock time limit in seconds. ``None`` (default)
+                disables the time limit.
             lam_prox: Trust region (proximal) weight. Either a float
                 (applied uniformly to all states and controls) or a dict
                 mapping state/control names to weights, e.g.
@@ -166,6 +169,7 @@ class PenalizedTrustRegion(Algorithm):
 
         # SCP convergence parameters
         self.k_max = k_max
+        self.t_max = t_max
         self.ep_tr = ep_tr
         self.ep_vb = ep_vb
         self.ep_vc = ep_vc

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170}/openscvx/problem.py

@@ -972,12 +972,18 @@ class Problem:
         }
 
     def solve(
-        self, max_iters: Optional[int] = None, continuous: bool = False
+        self,
+        max_iters: Optional[int] = None,
+        time_limit: Optional[float] = None,
+        continuous: bool = False,
     ) -> OptimizationResults:
         """Run the SCP algorithm until convergence or iteration limit.
 
         Args:
             max_iters: Maximum iterations (default: algorithm.k_max)
+            time_limit: Wall-clock time limit in seconds. Overrides
+                ``algorithm.t_max`` when provided. ``None`` (default) falls
+                back to ``algorithm.t_max``.
             continuous: If True, run all iterations regardless of convergence
 
         Returns:
@@ -1007,11 +1013,14 @@ class Problem:
             printing.header(self._columns)
 
         k_max = max_iters if max_iters is not None else self._algorithm.k_max
+        t_max = time_limit if time_limit is not None else self._algorithm.t_max
 
         while self._state.k <= k_max:
             result = self.step()
             if result["converged"] and not continuous:
                 break
+            if t_max is not None and (time.time() - t_0_while) >= t_max:
+                break
 
         t_f_while = time.time()
         self.timing_solve = t_f_while - t_0_while
@@ -1030,7 +1039,8 @@ class Problem:
         # Store solution state
         self._solution = copy.deepcopy(self._state)
 
-        return self._format_result(self._state, self._state.k <= k_max)
+        timed_out = t_max is not None and self.timing_solve >= t_max
+        return self._format_result(self._state, self._state.k <= k_max and not timed_out)
 
     def post_process(self) -> OptimizationResults:
         """Propagate solution through full nonlinear dynamics for high-fidelity trajectory.

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170}/openscvx/solvers/base.py

@@ -245,7 +245,14 @@ _SOLVER_MAP: Dict[str, type] = {}  # populated by __init__.py after all classes
 
 
 class SolverSpec(BaseModel):
-    """Validates solver configuration from dict/YAML input."""
+    """Validates solver configuration from dict/YAML input.
+
+    !!! warning
+        Enabling ``cvxpygen`` currently disables sparse parameter declarations.
+        cvxpygen does not yet support the N-D sparsity indices used by
+        OpenSCvx's tiled parameters, so all parameters are created as dense
+        when code generation is active.
+    """
 
     type: Literal["PTRSolver"] = "PTRSolver"
     cvx_solver: str = "QOCO"

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170}/openscvx/solvers/ptr_solver.py

@@ -119,6 +119,14 @@ class PTRSolver(ConvexSolver):
             ``{"abstol": 1e-6, "reltol": 1e-9, "enforce_dpp": True}``.
         cvxpygen: Enable CVXPy code generation for faster solves.
             Defaults to ``False``.
+
+            !!! warning
+                Enabling cvxpygen currently disables sparse parameter
+                declarations. cvxpygen does not yet support the N-D sparsity
+                indices used by OpenSCvx's tiled parameters, so all parameters
+                are created as dense when code generation is active. This may
+                increase the generated solver's memory footprint and compile
+                time but does not affect solution correctness.
         cvxpygen_override: Overwrite existing generated solver directory
             without prompting. Defaults to ``False``.
 
@@ -234,6 +242,14 @@ class PTRSolver(ConvexSolver):
             B_d_sp = _tile_sparsity(B_d_pat, N - 1)
             C_d_sp = _tile_sparsity(C_d_pat, N - 1)
 
+        # TODO: (griffin-norris) Remove once cvxpygen supports N-D sparsity
+        # indices. cvxpygen's handle_sparsity() assumes 2-D (rows, cols) but
+        # our tiled parameters produce 3-D indices (slices, rows, cols).
+        # Dropping sparsity here is safe — it only affects codegen performance.
+        if self.cvxpygen:
+            A_d_sp = B_d_sp = C_d_sp = None
+            constraint_sparsity = None
+
         # Create all CVXPy variables for the OCP
         self._ocp_vars = create_cvxpy_variables(
             N=N,

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170/openscvx.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openscvx
-Version: 0.4.1.dev167
+Version: 0.4.1.dev170
 Summary: A general Python-based successive convexification implementation which uses a JAX backend.
 Author-email: Chris Hayner and Griffin Norris <haynec@uw.edu>
 License: Apache Software License

{openscvx-0.4.1.dev167 → openscvx-0.4.1.dev170}/tests/test_brachistochrone.py

@@ -76,7 +76,7 @@ def _assert_brachistochrone_accuracy(comparison, problem, result):
         assert num_iters < 27, f"Took {num_iters} SCP iterations (expected < 15)"
 
     # Check timing - these are generous limits for a simple problem like brachistochrone
-    assert problem.timing_init < 12.0, (
+    assert problem.timing_init < 15.0, (
         f"Initialization took {problem.timing_init:.2f}s (expected < 12s)"
     )
     assert problem.timing_solve < 1.2, f"Solve took {problem.timing_solve:.2f}s (expected < 1.2s)"