bayinx 0.2.13__tar.gz → 0.2.19__tar.gz

{bayinx-0.2.13 → bayinx-0.2.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bayinx
-Version: 0.2.13
+Version: 0.2.19
 Summary: Bayesian Inference with JAX
 Requires-Python: >=3.12
 Requires-Dist: equinox>=0.11.12

bayinx-0.2.19/pyproject.toml

@@ -0,0 +1,44 @@
+[project]
+name = "bayinx"
+version = "0.2.19"
+description = "Bayesian Inference with JAX"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "equinox>=0.11.12",
+    "jax>=0.4.38",
+    "jaxtyping>=0.2.36",
+    "optax>=0.2.4",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.pytest.ini_options]
+addopts = "-q --benchmark-min-rounds=30 --benchmark-columns=rounds,mean,median,stddev --benchmark-group-by=func"
+
+[tool.bumpversion]
+current_version = "0.2.19"
+parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
+serialize = ["{major}.{minor}.{patch}"]
+search = "{current_version}"
+replace = "{new_version}"
+regex = false
+ignore_missing_version = false
+ignore_missing_files = false
+tag = true
+sign_tags = false
+tag_name = "v{new_version}"
+tag_message = "Bump version: {current_version} → {new_version}"
+allow_dirty = false
+commit = false
+message = "Bump version: {current_version} → {new_version}"
+moveable_tags = []
+commit_args = ""
+setup_hooks = []
+pre_commit_hooks = []
+post_commit_hooks = []
+
+[dependency-groups]
+dev = ["matplotlib>=3.10.1", "pytest>=8.3.5", "pytest-benchmark>=5.1.0"]

{bayinx-0.2.13 → bayinx-0.2.19}/src/bayinx/dists/normal.py

@@ -19,7 +19,7 @@ def prob(
     The PDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `mu`, and `sigma`.
     """
 
-    return _lax.exp(-0.5 * _lax.square((x - mu) / sigma)) / (
+    return _lax.exp(-0.5 * _lax.square((x - mu) / sigma)) / ( # pyright: ignore
         sigma * _lax.sqrt(2.0 * _PI)
     )
 
@@ -39,7 +39,7 @@ def logprob(
     The log of the PDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `mu`, and `sigma`.
     """
 
-    return -_lax.log(sigma * _lax.sqrt(2.0 * _PI)) - 0.5 * _lax.square((x - mu) / sigma)
+    return -_lax.log(sigma * _lax.sqrt(2.0 * _PI)) - 0.5 * _lax.square((x - mu) / sigma) # pyright: ignore
 
 
 def uprob(
@@ -57,7 +57,7 @@ def uprob(
     The uPDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `mu`, and `sigma`.
     """
 
-    return _lax.exp(-0.5 * _lax.square((x - mu) / sigma)) / sigma
+    return _lax.exp(-0.5 * _lax.square((x - mu) / sigma)) / sigma # pyright: ignore
 
 
 def ulogprob(
@@ -75,4 +75,4 @@ def ulogprob(
     The log uPDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `mu`, and `sigma`.
     """
 
-    return -_lax.log(sigma) - 0.5 * _lax.square((x - mu) / sigma)
+    return -_lax.log(sigma) - 0.5 * _lax.square((x - mu) / sigma) # pyright: ignore

{bayinx-0.2.13 → bayinx-0.2.19}/src/bayinx/mhx/vi/normalizing_flow.py

@@ -1,8 +1,6 @@
-from functools import partial
 from typing import Any, Callable, Self, Tuple
 
 import equinox as eqx
-import jax
 import jax.flatten_util as jfu
 import jax.numpy as jnp
 import jax.random as jr
@@ -61,7 +59,6 @@ class NormalizingFlow(Variational):
         return draws
 
     @eqx.filter_jit
-    @partial(jax.vmap, in_axes=(None, 0))
     def eval(self, draws: Array) -> Array:
         # Evaluate base density
         variational_evals: Array = self.base.eval(draws)

{bayinx-0.2.13 → bayinx-0.2.19}/src/bayinx/mhx/vi/standard.py

@@ -52,7 +52,7 @@ class Standard(Variational):
             x=draws,
             mu=jnp.array(0.0),
             sigma=jnp.array(1.0),
-        ).sum(axis=1)
+        ).sum(axis=1, keepdims=True)
 
     @eqx.filter_jit
     def filter_spec(self):

bayinx-0.2.13/pyproject.toml

@@ -1,22 +0,0 @@
-[project]
-name = "bayinx"
-version = "0.2.13"
-description = "Bayesian Inference with JAX"
-readme = "README.md"
-requires-python = ">=3.12"
-dependencies = [
-    "equinox>=0.11.12",
-    "jax>=0.4.38",
-    "jaxtyping>=0.2.36",
-    "optax>=0.2.4",
-]
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[tool.pytest.ini_options]
-addopts = "-q --benchmark-min-rounds=30 --benchmark-columns=rounds,mean,median,stddev --benchmark-group-by=func"
-
-[dependency-groups]
-dev = ["matplotlib>=3.10.1", "pytest>=8.3.5", "pytest-benchmark>=5.1.0"]