bayinx 0.2.32__tar.gz → 0.2.33__tar.gz

{bayinx-0.2.32 → bayinx-0.2.33}/PKG-INFO

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

{bayinx-0.2.32 → bayinx-0.2.33}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bayinx"
-version = "0.2.32"
+version = "0.2.33"
 description = "Bayesian Inference with JAX"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -19,7 +19,7 @@ build-backend = "hatchling.build"
 addopts = "-q --benchmark-min-rounds=30 --benchmark-columns=rounds,mean,median,stddev --benchmark-group-by=func"
 
 [tool.bumpversion]
-current_version = "0.2.32"
+current_version = "0.2.33"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 search = "{current_version}"

{bayinx-0.2.32 → bayinx-0.2.33}/src/bayinx/core/model.py

@@ -4,7 +4,7 @@ from typing import Any, Dict, Tuple
 import equinox as eqx
 import jax.numpy as jnp
 import jax.tree as jt
-from jaxtyping import Array, Scalar
+from jaxtyping import Array, PyTree, Scalar
 
 from bayinx.core.constraint import Constraint
 
@@ -18,7 +18,7 @@ class Model(eqx.Module):
     - `constraints`: A dictionary of constraints.
     """
 
-    params: Dict[str, Array]
+    params: Dict[str, PyTree]
     constraints: Dict[str, Constraint]
 
     @abstractmethod