Skip to content

Commit c4659b4

Browse files
committed
Simplify calculation of identity scaling matrix
1 parent 52761c2 commit c4659b4

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

pymc3/step_methods/hmc/base_hmc.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
from ..arraystep import ArrayStepShared
2-
from ...blocking import DictToArrayBijection, ArrayOrdering
32
from .trajectory import get_theano_hamiltonian_functions
43

54
from pymc3.tuning import guess_scaling
@@ -43,8 +42,7 @@ def __init__(self, vars=None, scaling=None, step_scale=0.25, is_cov=False,
4342
vars = inputvars(vars)
4443

4544
if scaling is None and potential is None:
46-
bij = DictToArrayBijection(ArrayOrdering(vars), model.test_point)
47-
scaling = np.ones(bij.map(model.test_point).size)
45+
scaling = np.ones(model.dict_to_array(model.test_point).size)
4846

4947
if isinstance(scaling, dict):
5048
scaling = guess_scaling(Point(scaling, model=model), model=model, vars=vars)

0 commit comments

Comments
 (0)