From fe7d151bc1c07ef36fadc694c443b1dfe9a73d06 Mon Sep 17 00:00:00 2001 From: jeandut Date: Mon, 11 Mar 2024 14:26:07 +0100 Subject: [PATCH] fixing API --- fedeca/fedeca_core.py | 5 +++-- fedeca/utils/experiment_utils.py | 7 ++----- 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/fedeca/fedeca_core.py b/fedeca/fedeca_core.py index 79fd3eb8..2bdd787b 100644 --- a/fedeca/fedeca_core.py +++ b/fedeca/fedeca_core.py @@ -844,8 +844,9 @@ def print_summary(self): self.get_final_cox_model() logger.info("Computing summary...") self.compute_summary() - logger.info("Final partial log-likelihood.s:") - logger.info(self.lls) + logger.info("Final partial log-likelihood:") + self.log_likelihood_ = self.lls[0] + logger.info(self.log_likelihood_) logger.info(self.results_) def get_final_cox_model(self): diff --git a/fedeca/utils/experiment_utils.py b/fedeca/utils/experiment_utils.py index 630aea7a..a55f4503 100644 --- a/fedeca/utils/experiment_utils.py +++ b/fedeca/utils/experiment_utils.py @@ -239,12 +239,9 @@ def single_experiment( .transpose() .add_prefix("smd_weighted_") ) - + breakpoint() # Check special case of FedECA - if ( - log_likelihood := getattr(model, "ll", None) # noqa: E231, E999, E251 - ) is None: - log_likelihood = model.lls[0] + log_likelihood = model.log_likelihood_ if name != "FedECA": backend_type = "N/A"