Update blackjax/mcmc/metrics.py

Co-authored-by: Junpeng Lao <junpenglao@gmail.com>
blackjax-devs · Sep 16, 2024 · c8ee838 · c8ee838
1 parent 017a85a
commit c8ee838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.