bayinx 0.3.18__py3-none-any.whl → 0.3.19__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.18.dist-info → bayinx-0.3.19.dist-info}/METADATA +1 -1
  3. {bayinx-0.3.18.dist-info → bayinx-0.3.19.dist-info}/RECORD +5 -5
  4. {bayinx-0.3.18.dist-info → bayinx-0.3.19.dist-info}/WHEEL +0 -0
  5. {bayinx-0.3.18.dist-info → bayinx-0.3.19.dist-info}/licenses/LICENSE +0 -0
bayinx/core/_variational.py CHANGED
@@ -198,7 +198,7 @@ class Variational(eqx.Module, Generic[M]):
198
198
  next, key = jr.split(key)
199
199
 
200
200
  # Draw from variational
201
- draw: Array = self.sample(1, key)
201
+ draw: Array = self.sample(1, key)[0]
202
202
 
203
203
  # Reconstruct model
204
204
  model: M = self.reconstruct_model(draw)
{bayinx-0.3.18.dist-info → bayinx-0.3.19.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: bayinx
3
- Version: 0.3.18
3
+ Version: 0.3.19
4
4
  Summary: Bayesian Inference with JAX
5
5
  License-File: LICENSE
6
6
  Requires-Python: >=3.12
{bayinx-0.3.18.dist-info → bayinx-0.3.19.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=NlCvBHsfP-7HeNimhD4x9JLT5VWeiCBUADr3bx6tt84,6381
11
+ bayinx/core/_variational.py,sha256=ljhLi9vbw8wh-z0Eisf0C08IoKlgwpx0VONh_ES-HmI,6384
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.18.dist-info/METADATA,sha256=osJ-IhnQs0OuZoA6AGk0GKSwibISSNaOeyLg5F0T4w4,3087
35
- bayinx-0.3.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
36
- bayinx-0.3.18.dist-info/licenses/LICENSE,sha256=VMhLhj5hx6VAENZBaNfXrmsNl7ov9uRh0jZ6D3ltgv4,1070
37
- bayinx-0.3.18.dist-info/RECORD,,
34
+ bayinx-0.3.19.dist-info/METADATA,sha256=V1qTM781r9t4cDGSRkgvn4PQAoxtsD9cFrMj0YVhUHo,3087
35
+ bayinx-0.3.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
36
+ bayinx-0.3.19.dist-info/licenses/LICENSE,sha256=VMhLhj5hx6VAENZBaNfXrmsNl7ov9uRh0jZ6D3ltgv4,1070
37
+ bayinx-0.3.19.dist-info/RECORD,,