Make Nprobes=1 possible

415558f2 · Philipp Arras · 28684298 · 415558f2
Commit 415558f2 authored Feb 05, 2018 by Philipp Arras
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

nifty4/probing/utils.py nifty4/probing/utils.py +3 -0

No files found.
--- a/nifty4/probing/utils.py
+++ b/nifty4/probing/utils.py
@@ -52,4 +52,7 @@ def probe_with_posterior_samples(op, post_op, nprobes):
    for i in range(nprobes):
        sample = post_op(op.generate_posterior_sample())
        sc.add(sample)
+
+    if nprobes == 1:
+        return sc.mean, None
    return sc.mean, sc.var