Skip to content

Commit e612e0c

Browse files
committed
Ignore zip/numba issue
1 parent 0df6d5b commit e612e0c

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

src/tranquilo/subsolvers/bntr_fast.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
"""Auxiliary functions for the quadratic BNTR trust-region subsolver."""
2+
23
import numpy as np
4+
from numba import njit
5+
36
from tranquilo.subsolvers._conjugate_gradient_fast import (
47
minimize_trust_cg_fast,
58
)
@@ -9,7 +12,6 @@
912
from tranquilo.subsolvers._trsbox_fast import (
1013
minimize_trust_trsbox_fast,
1114
)
12-
from numba import njit
1315

1416
EPSILON = np.finfo(float).eps ** (2 / 3)
1517

@@ -790,8 +792,7 @@ def _perform_gradient_descent_step(
790792
square_terms = x_inactive.T @ hessian_inactive @ x_inactive
791793

792794
predicted_reduction = trustregion_radius * (
793-
gradient_norm
794-
- 0.5 * trustregion_radius * square_terms / (gradient_norm**2)
795+
gradient_norm - 0.5 * trustregion_radius * square_terms / (gradient_norm**2)
795796
)
796797
actual_reduction = f_candidate_initial - f_candidate
797798

@@ -1111,7 +1112,9 @@ def _update_trustregion_radius_and_gradient_descent(
11111112
def _get_fischer_burmeister_direction_vector(x, gradient, lower_bounds, upper_bounds):
11121113
"""Compute the constrained direction vector via the Fischer-Burmeister function."""
11131114
direction = np.zeros(len(x))
1114-
for i, (x_, g_, l_, u_) in enumerate(zip(x, gradient, lower_bounds, upper_bounds)):
1115+
for i, (x_, g_, l_, u_) in enumerate(
1116+
zip(x, gradient, lower_bounds, upper_bounds) # noqa: B905 # strict=... not supported by numba yet, see https://github.com/numba/numba/issues/8943
1117+
):
11151118
fischer_scalar = _get_fischer_burmeister_scalar(u_ - x_, -g_)
11161119
fischer_scalar = _get_fischer_burmeister_scalar(fischer_scalar, x_ - l_)
11171120

0 commit comments

Comments
 (0)