bayinx 0.2.31__py3-none-any.whl → 0.2.32__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 (4) hide show
  1. bayinx/mhx/vi/flows/fullaffine.py +1 -1
  2. {bayinx-0.2.31.dist-info → bayinx-0.2.32.dist-info}/METADATA +1 -1
  3. {bayinx-0.2.31.dist-info → bayinx-0.2.32.dist-info}/RECORD +4 -4
  4. {bayinx-0.2.31.dist-info → bayinx-0.2.32.dist-info}/WHEEL +0 -0
bayinx/mhx/vi/flows/fullaffine.py CHANGED
@@ -40,7 +40,7 @@ class FullAffine(Flow):
40
40
  diag: Array = jnp.exp(jnp.diag(scale))
41
41
 
42
42
  # Return matrix with modified diagonal
43
- return jnp.fill_diagonal(scale, diag, inplace=False)
43
+ return jnp.fill_diagonal(jnp.tril(scale), diag, inplace=False)
44
44
 
45
45
  self.constraints = {"scale": constrain_scale}
46
46
 
{bayinx-0.2.31.dist-info → bayinx-0.2.32.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: bayinx
3
- Version: 0.2.31
3
+ Version: 0.2.32
4
4
  Summary: Bayesian Inference with JAX
5
5
  Requires-Python: >=3.12
6
6
  Requires-Dist: equinox>=0.11.12
{bayinx-0.2.31.dist-info → bayinx-0.2.32.dist-info}/RECORD RENAMED
@@ -20,10 +20,10 @@ bayinx/mhx/vi/meanfield.py,sha256=BobfTagVGA5R-dclv-E0jSA80KZg1X6GGjiw7XR61vE,36
20
20
  bayinx/mhx/vi/normalizing_flow.py,sha256=DYhvTiu2Fr5x8KpWAMZVUaio7ctG2X2SMUO0l8zfZ5g,4622
21
21
  bayinx/mhx/vi/standard.py,sha256=HaJsIz70Qo1Ql2hMQ-GQhcnfWiOGtyxgkOsm_yQaDKI,1718
22
22
  bayinx/mhx/vi/flows/__init__.py,sha256=Hn0Wqvvyv8Vr-mFmimwgNKCByxj-fjrlIvdR7tUSolg,180
23
- bayinx/mhx/vi/flows/fullaffine.py,sha256=s-fxgzv84BEqNMnmLt6vtvwJqBzixCC2OwWXLz3IK-w,1940
23
+ bayinx/mhx/vi/flows/fullaffine.py,sha256=Z_G2Cg90Asgvqel8buMx5kEVsiIxDDh8rc-L_nP9OCY,1950
24
24
  bayinx/mhx/vi/flows/planar.py,sha256=WVj-oxcRctuoRA6KJjU63ek1ZgKNG2vI-TLN0QqjtKA,1916
25
25
  bayinx/mhx/vi/flows/radial.py,sha256=Obj3SraliawIHmP14F9wRpWt34y3kscY--Izy24eCvM,2499
26
26
  bayinx/mhx/vi/flows/sylvester.py,sha256=ppK0BmS_ThvrCEhJiP_-p-kj67TQHSlU_RUZpDbIhsQ,469
27
- bayinx-0.2.31.dist-info/METADATA,sha256=ec8tXMczZabIJppu7h8LUdTS4D_TitqeS2b5ue1HAYI,3058
28
- bayinx-0.2.31.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
29
- bayinx-0.2.31.dist-info/RECORD,,
27
+ bayinx-0.2.32.dist-info/METADATA,sha256=C6xrmvyGJ573nlEObTgIb_CWiPDOY0zAtQj75qbgCc4,3058
28
+ bayinx-0.2.32.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
29
+ bayinx-0.2.32.dist-info/RECORD,,