use durations=0

fehiepsi · fehiepsi · commit 1abafd1e01c5 · 2020-07-12T22:47:11.000-05:00
to measure times
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -7,7 +7,7 @@ on:
     branches: [ master ]
 
 jobs:
-  chapters_00_13:
+  even_chapters:
 
     runs-on: ubuntu-latest
     strategy:
@@ -28,11 +28,10 @@ jobs:
         pip install nbval
     - name: Test with nbval
       run: |
-        find notebooks -maxdepth 1 -name "0*.ipynb" | sort -n | xargs pytest -vx --nbval-lax
-        find notebooks -maxdepth 1 -name "1[0-3]*.ipynb" | sort -n | xargs pytest -vx --nbval-lax
+        find notebooks -maxdepth 1 -name "[01][02468]*.ipynb" | sort -n | xargs pytest -vx --nbval-lax --durations=0
 
 
-  chapters_14_17:
+  odd_chapters:
 
     runs-on: ubuntu-latest
     strategy:
@@ -53,5 +52,5 @@ jobs:
         pip install nbval
     - name: Test with nbval
       run: |
-        find notebooks -maxdepth 1 -name "1[4-7]*.ipynb" | sort -n | xargs pytest -vx --nbval-lax
+        find notebooks -maxdepth 1 -name "[01][13579]*.ipynb" | sort -n | xargs pytest -vx --nbval-lax --durations=0
 
diff --git a/notebooks/13_models_with_memory.ipynb b/notebooks/13_models_with_memory.ipynb
@@ -15952,7 +15952,7 @@
     "    b = numpyro.sample(\"b\", dist.Normal(0, 0.5), sample_shape=(4,))\n",
     "    logit_p = a[actor] + g[block_id] + b[treatment]\n",
     "    if link:\n",
-    "        numpyro.sample(\"p\", expit(logit_p))\n",
+    "        numpyro.deterministic(\"p\", expit(logit_p))\n",
     "    numpyro.sample(\"pulled_left\", dist.Binomial(logits=logit_p), obs=pulled_left)\n",
     "\n",
     "\n",
