Skip to content

Commit 1aea645

Browse files
committed
Revert "Merge pull request #2232 from a-rodin/scaling-default-to-identity"
This reverts commit eb6042f, reversing changes made to c3afa00.
1 parent eb6042f commit 1aea645

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

pymc3/step_methods/hmc/base_hmc.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from pymc3.model import modelcontext, Point
66
from .quadpotential import quad_potential
77
from pymc3.theanof import inputvars, make_shared_replacements
8-
import numpy as np
98

109

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

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

4746
if isinstance(scaling, dict):
4847
scaling = guess_scaling(Point(scaling, model=model), model=model, vars=vars)

0 commit comments

Comments
 (0)