fixing API

owkin · Mar 11, 2024 · fe7d151 · fe7d151
1 parent 99fc03a
commit fe7d151
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git 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
@@ -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"