bayinx 0.3.16__py3-none-any.whl → 0.3.17__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (5) hide show
  1. bayinx/core/_variational.py +1 -1
  2. {bayinx-0.3.16.dist-info → bayinx-0.3.17.dist-info}/METADATA +1 -1
  3. {bayinx-0.3.16.dist-info → bayinx-0.3.17.dist-info}/RECORD +5 -5
  4. {bayinx-0.3.16.dist-info → bayinx-0.3.17.dist-info}/WHEEL +0 -0
  5. {bayinx-0.3.16.dist-info → bayinx-0.3.17.dist-info}/licenses/LICENSE +0 -0
bayinx/core/_variational.py CHANGED
@@ -176,7 +176,7 @@ class Variational(eqx.Module, Generic[M]):
176
176
  key: Key = jr.PRNGKey(0),
177
177
  ) -> Array:
178
178
  # Sample a single draw to evaluate shape of output
179
- draw: Array = self.sample(1, key)
179
+ draw: Array = self.sample(1, key)[0]
180
180
  output: Array = func(self._unflatten(draw), data)
181
181
 
182
182
  # Allocate space for results
{bayinx-0.3.16.dist-info → bayinx-0.3.17.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: bayinx
3
- Version: 0.3.16
3
+ Version: 0.3.17
4
4
  Summary: Bayesian Inference with JAX
5
5
  License-File: LICENSE
6
6
  Requires-Python: >=3.12
{bayinx-0.3.16.dist-info → bayinx-0.3.17.dist-info}/RECORD RENAMED
@@ -8,7 +8,7 @@ bayinx/core/_flow.py,sha256=3q4rKvATnbUpuj4ICUd4lIxu_3z7GRDuNujVhAku1X0,2342
8
8
  bayinx/core/_model.py,sha256=FJUyYVE9e2uTFamxtSMKY_VV2stiU2QF68Wl_7EAKEU,2895
9
9
  bayinx/core/_optimization.py,sha256=mmeVUqfFARz8F7q4LRl-uEwVWzekmzh-9o7PnuvsHZk,2651
10
10
  bayinx/core/_parameter.py,sha256=r20JedTW2lY0miNNh9y6LeIVAsGX1kP_rlGxphW_jZg,1080
11
- bayinx/core/_variational.py,sha256=MbOgWs0-mkEwIYsh1Hhn7anWkQ7fAGjkr6VJybYAuR0,6185
11
+ bayinx/core/_variational.py,sha256=mLumWRcpATuU-S073qFy-6oYL-OpI3wNrFe7bXEbyRE,6188
12
12
  bayinx/dists/__init__.py,sha256=BIrypqMnTLWK3a_zw8fYKMyuEMxP_qGsLfLeScias0o,118
13
13
  bayinx/dists/bernoulli.py,sha256=xMV9BgtVX_1XkPdZ43q0meMIEkgMyuUPx--dyo6_DKs,1006
14
14
  bayinx/dists/gamma2.py,sha256=HtB60LUQdPj4yDAHme2jsHNmLfrAKWsSZnDYkxAGaOI,1548
@@ -31,7 +31,7 @@ bayinx/mhx/vi/flows/fullaffine.py,sha256=11y_A0oO3bkKDSz-EQ6Sf4Ec2M7vHZxw94EdvAD
31
31
  bayinx/mhx/vi/flows/planar.py,sha256=2I2WzIskl8MRpJkK13FQE3vSF-077qo8gRed_EL1Pn8,1920
32
32
  bayinx/mhx/vi/flows/radial.py,sha256=e0GfuO-CL8SVr3YnEfsxStpyKcJlFpzMyjMp3sa38hg,2503
33
33
  bayinx/mhx/vi/flows/sylvester.py,sha256=ppK0BmS_ThvrCEhJiP_-p-kj67TQHSlU_RUZpDbIhsQ,469
34
- bayinx-0.3.16.dist-info/METADATA,sha256=9oLNZAc1YvN9FpWiuFFlztL-OkhclOYPBQFkuiHmUT8,3087
35
- bayinx-0.3.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
36
- bayinx-0.3.16.dist-info/licenses/LICENSE,sha256=VMhLhj5hx6VAENZBaNfXrmsNl7ov9uRh0jZ6D3ltgv4,1070
37
- bayinx-0.3.16.dist-info/RECORD,,
34
+ bayinx-0.3.17.dist-info/METADATA,sha256=8YZLse5eUFxxJYvZ8lBHMs-5FbnxxkLTe1hEN4pxuz4,3087
35
+ bayinx-0.3.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
36
+ bayinx-0.3.17.dist-info/licenses/LICENSE,sha256=VMhLhj5hx6VAENZBaNfXrmsNl7ov9uRh0jZ6D3ltgv4,1070
37
+ bayinx-0.3.17.dist-info/RECORD,,