update notebooks

ComputationalPsychiatry · Sep 21, 2023 · e7f694a · e7f694a
1 parent 829316f
commit e7f694a
Show file tree

Hide file tree

Showing 4 changed files with 88 additions and 101 deletions.
diff --git a/docs/source/notebooks/1.1-Binary_HGF.ipynb b/docs/source/notebooks/1.1-Binary_HGF.ipynb
diff --git a/docs/source/notebooks/1.1-Binary_HGF.md b/docs/source/notebooks/1.1-Binary_HGF.md
@@ -214,7 +214,8 @@ with pm.Model() as two_levels_binary_hgf:
         hgf_logp_op(
             omega_1=jnp.inf,
             omega_2=omega_2,
-            continuous_precision=jnp.inf,
+            continuous_precision=jnp.nan,
+            binary_precision=jnp.inf,
             rho_1=0.0,
             rho_2=0.0,
             pi_1=0.0,
@@ -315,7 +316,8 @@ with pm.Model() as three_levels_binary_hgf:
             omega_1=jnp.inf,
             omega_2=omega_2,
             omega_3=omega_3,
-            continuous_precision=jnp.inf,
+            continuous_precision=jnp.nan,
+            binary_precision=jnp.inf,
             rho_1=0.0,
             rho_2=0.0,
             rho_3=0.0,

diff --git a/docs/source/notebooks/1.3-Continuous_HGF.ipynb b/docs/source/notebooks/1.3-Continuous_HGF.ipynb
diff --git a/docs/source/notebooks/1.3-Continuous_HGF.md b/docs/source/notebooks/1.3-Continuous_HGF.md
@@ -5,7 +5,7 @@ jupytext:
     extension: .md
     format_name: myst
     format_version: 0.13
-    jupytext_version: 1.14.7
+    jupytext_version: 1.15.1
 kernelspec:
   display_name: Python 3 (ipykernel)
   language: python
@@ -225,6 +225,7 @@ with pm.Model() as two_level_hgf:
             omega_1=omega_1,
             omega_2=-2.0,
             continuous_precision=1e4,
+            binary_precision=jnp.nan,
             rho_1=0.0,
             rho_2=0.0,
             pi_1=1e4,
@@ -313,6 +314,7 @@ with pm.Model() as three_level_hgf:
             omega_2=-2.0,
             omega_3=-2.0,
             continuous_precision=1e4,
+            binary_precision=jnp.nan,
             rho_1=0.0,
             rho_2=0.0,
             rho_3=0.0,