From 7c09ea1382798f2178d93d7438b747c74b15b0ed Mon Sep 17 00:00:00 2001 From: Gregory Ashton <gregory.ashton@ligo.org> Date: Tue, 17 Oct 2017 14:30:28 +0200 Subject: [PATCH] Minor polishing to the slide window method --- pyfstat/grid_based_searches.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/pyfstat/grid_based_searches.py b/pyfstat/grid_based_searches.py index 2a9d098..0e16391 100644 --- a/pyfstat/grid_based_searches.py +++ b/pyfstat/grid_based_searches.py @@ -346,7 +346,8 @@ class SliceGridSearch(GridSearch): raise ValueError( 'Lambda0 must be of length {}'.format(len(self.search_keys))) - def run(self, factor=2, max_n_ticks=4, whspace=0.07, save=True): + def run(self, factor=2, max_n_ticks=4, whspace=0.07, save=True, + **kwargs): lbdim = 0.5 * factor # size of left/bottom margin trdim = 0.4 * factor # size of top/right margin plotdim = factor * self.ndim + factor * (self.ndim - 1.) * whspace @@ -409,8 +410,8 @@ class SliceGridSearch(GridSearch): search.set_out_file() search.run() ax = search.plot_2D(jkey, ikey, ax=ax, save=False, - y0=self.Lambda0[i], x0=self.Lambda0[j] - ) + y0=self.Lambda0[i], x0=self.Lambda0[j], + **kwargs) setattr(search, ikey+'s', [self.Lambda0[i]]) setattr(search, jkey+'s', [self.Lambda0[j]]) @@ -546,6 +547,7 @@ class FrequencySlidingWindow(GridSearch): self.F2s = [F2] self.Alphas = [Alpha] self.Deltas = [Delta] + self.input_arrays = False def inititate_search_object(self): logging.info('Setting up search object') -- GitLab