Skip to content

Commit 93cd831

Browse files
author
Martin Ingram
committed
Harmonize naming with existing code
1 parent 3b090ca commit 93cd831

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

pymc_extras/inference/dadvi/dadvi.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ def fit_dadvi(
2929
n_draws: int = 1000,
3030
keep_untransformed: bool = False,
3131
optimizer_method: minimize_method = "trust-ncg",
32-
use_jacobian: bool = True,
32+
use_grad: bool = True,
3333
use_hessp: bool = True,
3434
**minimize_kwargs,
3535
) -> az.InferenceData:
@@ -75,8 +75,9 @@ def fit_dadvi(
7575
``scipy.optimize.minimize`` function. See the documentation of
7676
that function for details.
7777
78-
use_jacobian:
79-
If True, pass the Jacobian function to `scipy.optimize.minimize`.
78+
use_grad:
79+
If True, pass the gradient function to
80+
`scipy.optimize.minimize` (where it is referred to as `jac`).
8081
8182
use_hessp:
8283
If True, pass the hessian vector product to `scipy.optimize.minimize`.
@@ -116,7 +117,7 @@ def fit_dadvi(
116117

117118
derivative_kwargs = {}
118119

119-
if use_jacobian:
120+
if use_grad:
120121
derivative_kwargs["jac"] = True
121122
if use_hessp:
122123
derivative_kwargs["hessp"] = f_hessp

0 commit comments

Comments
 (0)