Skip to content

Commit 6f58dbf

Browse files
a-rodintwiecki
authored andcommitted
Return identity matrix if no scaling provided (#2236)
1 parent 541afd8 commit 6f58dbf

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pymc3/step_methods/hmc/base_hmc.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
from pymc3.tuning import guess_scaling
55
from pymc3.model import modelcontext, Point
66
from .quadpotential import quad_potential
7-
from pymc3.theanof import inputvars, make_shared_replacements
7+
from pymc3.theanof import inputvars, make_shared_replacements, floatX
8+
import numpy as np
89

910

1011
class BaseHMC(ArrayStepShared):
@@ -41,7 +42,7 @@ def __init__(self, vars=None, scaling=None, step_scale=0.25, is_cov=False,
4142
vars = inputvars(vars)
4243

4344
if scaling is None and potential is None:
44-
scaling = model.test_point
45+
scaling = floatX(np.ones(model.dict_to_array(model.test_point).size))
4546

4647
if isinstance(scaling, dict):
4748
scaling = guess_scaling(Point(scaling, model=model), model=model, vars=vars)

0 commit comments

Comments
 (0)