diff --git a/pyfstat/grid_based_searches.py b/pyfstat/grid_based_searches.py index ea078e8d8b69d70207b2a995506a6527062e8241..6ed2963a740914b036a0738f0be18c4f38b976d6 100644 --- a/pyfstat/grid_based_searches.py +++ b/pyfstat/grid_based_searches.py @@ -399,7 +399,7 @@ class FrequencySlidingWindow(GridSearch): self.input_data = np.array(input_data) def plot_sliding_window(self, F0=None, ax=None, savefig=True, - colorbar=True): + colorbar=True, timestamps=False): data = self.data if ax is None: ax = plt.subplot() @@ -411,7 +411,7 @@ class FrequencySlidingWindow(GridSearch): dts = (tmids - self.minStartTime) / 86400. if F0: frequencies = frequencies - F0 - ax.set_ylabel('Frequency - $f_0$ [Hz]') + ax.set_ylabel('Frequency - $f_0$ [Hz] \n $f_0={:0.2f}$'.format(F0)) else: ax.set_ylabel('Frequency [Hz]') twoF = twoF.reshape((len(tmids), len(frequencies))) @@ -421,10 +421,17 @@ class FrequencySlidingWindow(GridSearch): cb = plt.colorbar(pax, ax=ax) cb.set_label('$2\mathcal{F}$') ax.set_xlabel( - r'Days from $t_\mathrm{{start}}$={}'.format(self.minStartTime)) + r'Mid-point (days after $t_\mathrm{{start}}$={})'.format( + self.minStartTime)) ax.set_title( 'Sliding window length = {} days in increments of {} days' - .format(self.window_size/86400, self.window_delta/86400)) + .format(self.window_size/86400, self.window_delta/86400), + ) + if timestamps: + axT = ax.twiny() + axT.set_xlim(tmids[0]*1e-9, tmids[-1]*1e-9) + axT.set_xlabel('Mid-point timestamp [GPS $10^{9}$ s]') + ax.set_title(ax.get_title(), y=1.18) if savefig: plt.tight_layout() plt.savefig( diff --git a/pyfstat/mcmc_based_searches.py b/pyfstat/mcmc_based_searches.py index 29892cc000208ca93867c4e5ecdc96a522b38829..9acd2fa341eaa2a4a9497bb9cb01ff12129b2c5c 100644 --- a/pyfstat/mcmc_based_searches.py +++ b/pyfstat/mcmc_based_searches.py @@ -544,6 +544,10 @@ class MCMCSearch(core.BaseSearchClass): """ + if 'truths' in kwargs and len(kwargs['truths']) != self.ndim: + logging.warning('len(Truths) != ndim, Truths will be ignored') + kwargs['truths'] = None + if self.ndim < 2: with plt.rc_context(rc_context): if fig_and_axes is None: