fixed summary bug

jzluo · Aug 7, 2022 · 08ccb05 · 08ccb05
1 parent c87868b
commit 08ccb05
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [0.5.0] - 2022-08-07
 ### Added
 - `test_vars` option to specify the variable(s) for which to calculate PL confidence intervals and p-values.
+### Fixed
+- Fixed bug where `.summary(xname)` would append `Intercept` to `xname` such that repeated calls would break.
 
 ## [0.4.0] - 2022-08-01
 ### Added

diff --git a/firthlogist/firthlogist.py b/firthlogist/firthlogist.py
@@ -244,6 +244,8 @@ def summary(self, xname=None, tablefmt="simple"):
         table += f"Log-Likelihood: {round(self.loglik_, 4)}\n"
         table += f"Newton-Raphson iterations: {self.n_iter_}\n"
         print(table)
+        if self.fit_intercept:
+            xname.pop()
         return
 
     def decision_function(self, X):