Merge pull request #209 from danielward27/fix_numerical_inverse_log_det

Fix numerical inverse log det
danielward27 · Feb 13, 2025 · 9c45347 · 9c45347
2 parents 5bcdc93 + 351148e
commit 9c45347
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flowjax/bijections/utils.py b/flowjax/bijections/utils.py
@@ -299,4 +299,4 @@ def transform_and_log_det(self, x, condition=None):
     def inverse_and_log_det(self, y, condition=None):
         x = self.inverter(self.bijection, y, condition)
         _, log_det = self.bijection.transform_and_log_det(x, condition)
-        return x, log_det
+        return x, -log_det
diff --git a/tests/test_bijections/test_bijections.py b/tests/test_bijections/test_bijections.py
@@ -212,7 +212,7 @@
         cond_shape=(),
     ),
     "NumericalInverse": lambda: NumericalInverse(
-        Affine(5),
+        Affine(5, 2),
         root_finder_to_inverter(
             partial(bisection_search, lower=-1, upper=1, atol=1e-7),
         ),