@@ -76,7 +76,7 @@ def get_internal_bounds(
7676 bounds : Bounds | None = None ,
7777 registry : PyTreeRegistry | None = None ,
7878 add_soft_bounds : bool = False ,
79- propagate_none : bool = True ,
79+ propagate_none_in_fast_path : bool = True ,
8080) -> tuple [NDArray [np .float64 ] | None , NDArray [np .float64 ] | None ]:
8181 """Create consolidated and flattened bounds for params.
8282
@@ -97,9 +97,9 @@ def get_internal_bounds(
9797 add_soft_bounds: If True, the element-wise maximum (minimum) of the lower and
9898 soft_lower (upper and soft_upper) bounds are taken. If False, the lower
9999 (upper) bounds are returned.
100- propagate_none : If True, None-valued lower and upper bounds are propagated to
101- the output. If False, None values are replaced with -np.inf for the lower
102- bound and np.inf for the upper bound.
100+ propagate_none_in_fast_path : If True, None-valued lower and upper bounds are
101+ propagated to the output in the fast path case . If False, None values are
102+ replaced with -np.inf for the lower bound and np.inf for the upper bound.
103103
104104 Returns:
105105 Consolidated and flattened lower_bounds.
@@ -117,7 +117,7 @@ def get_internal_bounds(
117117 return _get_fast_path_bounds (
118118 params = params ,
119119 bounds = bounds ,
120- propagate_none = propagate_none ,
120+ propagate_none = propagate_none_in_fast_path ,
121121 )
122122
123123 # In the slow path, None bounds are currently not propagated, but replaced with
0 commit comments