From c720aeaef64201f8ba361785addf18b998aa130e Mon Sep 17 00:00:00 2001
From: Philipp Frank <philipp@mpa-garching.mpg.de>
Date: Fri, 18 Nov 2022 09:15:25 -0600
Subject: [PATCH] make evidences an array

---
 demo_poisson.ipynb          |  8 ++++----
 demo_poisson_solution.ipynb | 10 +++++-----
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/demo_poisson.ipynb b/demo_poisson.ipynb
index c817ad6..72eedb4 100644
--- a/demo_poisson.ipynb
+++ b/demo_poisson.ipynb
@@ -28,7 +28,7 @@
     "from utils import plot_2D, load_psf, geovi_sampling, plot_posterior\n",
     "\n",
     "ift.random.push_sseq_from_seed(42)\n",
-    "evidences = []"
+    "evidences = np.zeros(3)"
    ]
   },
   {
@@ -74,7 +74,7 @@
     "\n",
     "\n",
     "print(evidence)\n",
-    "evidences += [evidence, ]"
+    "evidences[0] = evidence"
    ]
   },
   {
@@ -88,7 +88,7 @@
     "# Inference model 2\n",
     "\n",
     "\n",
-    "evidences += [evidence, ]"
+    "evidences[1] = evidence"
    ]
   },
   {
@@ -149,7 +149,7 @@
    "source": [
     "# Inference model 3\n",
     "\n",
-    "evidences += [evidence, ]"
+    "evidences[2] = evidence"
    ]
   },
   {
diff --git a/demo_poisson_solution.ipynb b/demo_poisson_solution.ipynb
index 52055b8..2e0cbaf 100644
--- a/demo_poisson_solution.ipynb
+++ b/demo_poisson_solution.ipynb
@@ -28,7 +28,7 @@
     "from utils import plot_2D, load_psf, geovi_sampling, plot_posterior\n",
     "\n",
     "ift.random.push_sseq_from_seed(42)\n",
-    "evidences = []"
+    "evidences = np.zeros(3)"
    ]
   },
   {
@@ -84,7 +84,7 @@
     "samples, evidence = geovi_sampling(likelihood @ model1)\n",
     "plot_2D(samples.average(model1).val, 'model1 posterior mean')\n",
     "print(evidence)\n",
-    "evidences += [evidence, ]"
+    "evidences[0] = evidence"
    ]
   },
   {
@@ -100,7 +100,7 @@
     "samples, evidence = geovi_sampling(likelihood @ model2)\n",
     "plot_2D(samples.average(model2).val, 'model2 posterior mean')\n",
     "print(evidence)\n",
-    "evidences += [evidence, ]"
+    "evidences[1] = evidence"
    ]
   },
   {
@@ -110,7 +110,7 @@
    "outputs": [],
    "source": [
     "# Compare evidence\n",
-    "print(evidences)"
+    "print(evidences[:2])"
    ]
   },
   {
@@ -169,7 +169,7 @@
     "samples, evidence = geovi_sampling(likelihood @ model3)\n",
     "plot_2D(samples.average(model3).val, 'model3 posterior mean')\n",
     "print(evidence)\n",
-    "evidences += [evidence, ]"
+    "evidences[2] = evidence"
    ]
   },
   {
-- 
GitLab