Update blackjax/mcmc/metrics.py

make inv and trans required kwarg with type bool in metric.scale Co-authored-by: Junpeng Lao <[email protected]>
blackjax-devs · Oct 4, 2024 · d0a7066 · d0a7066
1 parent 230ff26
commit d0a7066
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/blackjax/mcmc/metrics.py b/blackjax/mcmc/metrics.py
@@ -65,8 +65,9 @@ def __call__(
         self,
         position: ArrayLikeTree,
         element: ArrayLikeTree,
-        inv: ArrayLikeTree,
-        trans: ArrayLikeTree,
+        *,
+        inv: bool,
+        trans: bool,
     ) -> ArrayLikeTree:
         ...