Bump version and ensure float params

danielward27 · Oct 7, 2024 · 14b7a52 · 14b7a52
1 parent e932a30
commit 14b7a52
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/flowjax/distributions.py b/flowjax/distributions.py
@@ -796,7 +796,10 @@ class Beta(AbstractDistribution):
     cond_shape: ClassVar[None] = None
 
     def __init__(self, alpha: ArrayLike, beta: ArrayLike):
-        alpha, beta = jnp.broadcast_arrays(alpha, beta)
+        alpha, beta = jnp.broadcast_arrays(
+            arraylike_to_array(alpha, dtype=float),
+            arraylike_to_array(beta, dtype=float),
+        )
         self.alpha = Parameterize(softplus, inv_softplus(alpha))
         self.beta = Parameterize(softplus, inv_softplus(beta))
         self.shape = alpha.shape

diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ license = { file = "LICENSE" }
 name = "flowjax"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "15.0.0"
+version = "15.1.0"
 
 [project.urls]
 repository = "https://github.com/danielward27/flowjax"