From 8fb9bf86007f48335645e58b1d536044c4af2ba1 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Thu, 25 Mar 2021 16:00:17 +0100 Subject: model: allow stats generation to be skipped --- lib/model.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'lib') diff --git a/lib/model.py b/lib/model.py index 8b76984..c89ff4f 100644 --- a/lib/model.py +++ b/lib/model.py @@ -69,6 +69,7 @@ class AnalyticModel: arg_count=None, function_override=dict(), use_corrcoef=False, + compute_stats=True, ): """ Create a new AnalyticModel and compute parameter statistics. @@ -119,7 +120,8 @@ class AnalyticModel: self.fit_done = False - self._compute_stats(by_name) + if compute_stats: + self._compute_stats(by_name) def __repr__(self): names = ", ".join(self.by_name.keys()) @@ -423,6 +425,7 @@ class PTAModel(AnalyticModel): use_corrcoef=False, pta=None, pelt=None, + compute_stats=True, ): """ Prepare a new PTA energy model. @@ -491,7 +494,8 @@ class PTAModel(AnalyticModel): self._aggregate_to_ndarray(self.by_name) - self._compute_stats(by_name) + if compute_stats: + self._compute_stats(by_name) if self.pelt is not None: # cluster_substates uses self.attr_by_name[*]["power"].param_values, which is set by _compute_stats -- cgit v1.2.3