9/n

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)