better-optimize 0.2.0__tar.gz → 0.2.1__tar.gz

{better_optimize-0.2.0 → better_optimize-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: better_optimize
-Version: 0.2.0
+Version: 0.2.1
 Summary: A drop-in replacement for scipy optimize functions with quality of life improvements
 License: MIT License
 Project-URL: Homepage, https://github.com/jessegrabowski/better_optimize

{better_optimize-0.2.0 → better_optimize-0.2.1}/better_optimize/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2025-12-13T05:36:24+0100",
+ "date": "2026-03-02T01:36:39+0100",
  "dirty": false,
  "error": null,
- "full-revisionid": "487a1c99b33650b7ba7264b19373ec2d54ca2e1f",
- "version": "0.2.0"
+ "full-revisionid": "14b8b75021dd22fa3d38135eee74b2ab95c30a06",
+ "version": "0.2.1"
 }
 '''  # END VERSION_JSON
 

{better_optimize-0.2.0 → better_optimize-0.2.1}/better_optimize/wrapper.py

@@ -4,6 +4,7 @@ from collections.abc import Callable
 from functools import partial
 
 import numpy as np
+import scipy.sparse as sp
 
 from rich.console import Console
 from rich.progress import (
@@ -134,10 +135,16 @@ class ObjectiveWrapper:
 
         value_dict = {"f_value": value}
         if grad is not None:
-            grad_norm = np.linalg.norm(grad)
+            if sp.issparse(grad):
+                grad_norm = sp.linalg.norm(grad)
+            else:
+                grad_norm = np.linalg.norm(grad)
             value_dict["grad_norm"] = grad_norm
         if hess is not None:
-            hess_norm = np.linalg.norm(hess)
+            if sp.issparse(hess):
+                hess_norm = sp.linalg.norm(hess)
+            else:
+                hess_norm = np.linalg.norm(hess)
             value_dict["hess_norm"] = hess_norm
 
         if self.n_eval == 0 and self.task is None:

{better_optimize-0.2.0 → better_optimize-0.2.1}/better_optimize.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: better_optimize
-Version: 0.2.0
+Version: 0.2.1
 Summary: A drop-in replacement for scipy optimize functions with quality of life improvements
 License: MIT License
 Project-URL: Homepage, https://github.com/jessegrabowski/better_optimize

{better_optimize-0.2.0 → better_optimize-0.2.1}/tests/test_wrapper.py

@@ -7,6 +7,7 @@ from functools import partial
 
 import numpy as np
 import pytest
+import scipy.sparse as sp
 
 from scipy.optimize import minimize as scipy_minimize
 from scipy.optimize import root as scipy_root
@@ -66,3 +67,32 @@ def test_execption_breaks_optimization(root, monkeypatch):
     # Non-KeyboardInterrupt exceptions should break execution.
     with pytest.raises(Exception, match="Simulated error"):
         optimizer_early_stopping_wrapper(f_optim)
+
+
+def func2(x, a, b):
+    f = [a * x[0] * np.cos(x[1]) - 4, x[1] * x[0] - b * x[1] - 5]
+
+    return np.array(f)
+
+
+def func2_sparse_jac(x, a, b):
+    df = np.array([[a * np.cos(x[1]), -a * x[0] * np.sin(x[1])], [x[1], x[0] - b]])
+    return sp.csr_matrix(df)
+
+
+def test_wrapper_compatible_with_sparse_outputs():
+    x0 = np.array([0.8, 0.8])
+
+    objective = ObjectiveWrapper(
+        f=partial(func2, a=1, b=1),
+        jac=partial(func2_sparse_jac, a=1, b=1),
+        maxeval=100,
+        progressbar=True,
+        root=True,
+    )
+
+    with objective.progress:
+        result = objective(x0)
+
+    value, grad = result
+    assert sp.issparse(grad)