diff --git a/helpers/plot.py b/helpers/plot.py index 03c452c7f68e10b6d828037d1b1288b28b43a233..d205bba9587391d549c562f4ee3030278c60b56a 100644 --- a/helpers/plot.py +++ b/helpers/plot.py @@ -92,7 +92,6 @@ def plot_prior_samples_2d(n_samps, signal, R, correlated_field, pspec, ss = ift.from_random(signal.domain) samples.append(ss) foo = pspec.force(ss).val - print(pspecmin, pspecmax) pspecmin = min([min(foo), pspecmin]) pspecmax = max([max(foo), pspecmin]) pspecmin /= 10 @@ -117,7 +116,7 @@ def plot_prior_samples_2d(n_samps, signal, R, correlated_field, pspec, raise ValueError('likelihood type not implemented') data = R.adjoint(data + 0.) - ax[ii, 0].plot(pspec.target[0].k_lengths, pspec.force(sample)) + ax[ii, 0].plot(pspec.target[0].k_lengths, pspec.force(sample).val) ax[ii, 0].set_yscale('log') ax[ii, 0].set_xscale('log') ax[ii, 0].set_ylim(pspecmin, pspecmax)