Skip to content

Commit 8b127a6

Browse files
committed
Don't cast float to np.array
1 parent bd939e0 commit 8b127a6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc/step_methods/hmc/nuts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,7 @@ def __init__(
257257
self.start = start
258258
self.step_size = step_size
259259
self.Emax = Emax
260-
self.start_energy = np.array(start.energy)
260+
self.start_energy = start.energy
261261

262262
self.left = self.right = start
263263
self.proposal = Proposal(start.q.data, start.q_grad, start.energy, start.model_logp, 0)

0 commit comments

Comments
 (0)