From 4340be165bd95f6c79141d1873dc512e22cb14d5 Mon Sep 17 00:00:00 2001
From: RunasSudo <runassudo@yingtongli.me>
Date: Sun, 16 Oct 2022 02:49:43 +1100
Subject: [PATCH] Fix RegressionResult.bayesfactor_beta_zero for categorical
 terms

---
 yli/regress.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/yli/regress.py b/yli/regress.py
index ba951ab..bd20806 100644
--- a/yli/regress.py
+++ b/yli/regress.py
@@ -145,10 +145,14 @@ class RegressionResult:
 		"""
 		Compute Bayes factor testing the hypothesis that the given beta is 0
 		Requires statsmodels regression
+		
+		term: Raw name (from statsmodels, available via RegressionResult.raw_result) of term to be tested
 		"""
 		
+		# FIXME: Allow specifying our renamed terms
+		
 		# Get parameters required for AFBF
-		params = pd.Series({term.raw_name.replace('[', '_').replace(']', '_'): term.beta.point for term in self.terms.values()})
+		params = pd.Series({raw_name.replace('[', '_').replace(']', '_'): beta for raw_name, beta in self.raw_result.params.items()})
 		cov = self.raw_result.cov_params()
 		
 		# Compute BF matrix