From c8ee8380770a991ec63454072ecd128d93901a75 Mon Sep 17 00:00:00 2001
From: Adrien Corenflos <adrien.corenflos@gmail.com>
Date: Mon, 16 Sep 2024 20:02:37 +0100
Subject: [PATCH] Update blackjax/mcmc/metrics.py

Co-authored-by: Junpeng Lao <junpenglao@gmail.com>
---
 blackjax/mcmc/metrics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/blackjax/mcmc/metrics.py b/blackjax/mcmc/metrics.py
index f7c4b25f8..8ce6c0e56 100644
--- a/blackjax/mcmc/metrics.py
+++ b/blackjax/mcmc/metrics.py
@@ -273,7 +273,7 @@ def is_turning(
         # return turning_at_left | turning_at_right
 
     def scale(
-        position: ArrayLikeTree, element: ArrayLikeTree, inv: ArrayLikeTree
+        position: ArrayLikeTree, element: ArrayLikeTree, inv: bool
     ) -> ArrayLikeTree:
         """Scale elements by the mass matrix.