From deb3a7d79e5ca0cc9250820a73e643ecaf995e13 Mon Sep 17 00:00:00 2001
From: Jakob Roth <roth@mpa-garching.mpg.de>
Date: Mon, 3 Mar 2025 15:49:17 +0100
Subject: [PATCH] hmc: use vdot instead of deprecated dot
---
src/re/hmc.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/re/hmc.py b/src/re/hmc.py
index fd7b00dd4..06f850d48 100644
--- a/src/re/hmc.py
+++ b/src/re/hmc.py
@@ -10,7 +10,7 @@ from jax.experimental import io_callback
from jax.scipy.special import expit
from .lax import cond, fori_loop, while_loop
-from .tree_math import random_like
+from .tree_math import random_like, vdot
_DEBUG_FLAG = False
@@ -405,8 +405,8 @@ def is_euclidean_uturn(qp_left, qp_right):
--------
Betancourt - A conceptual introduction to Hamiltonian Monte Carlo
"""
- return (qp_right.momentum.dot(qp_right.position - qp_left.position) < 0.0) & (
- qp_left.momentum.dot(qp_left.position - qp_right.position) < 0.0
+ return (vdot(qp_right.momentum, qp_right.position - qp_left.position) < 0.0) & (
+ vdot(qp_left.momentum, qp_left.position - qp_right.position) < 0.0
)
--
GitLab