From cc472acdeea4df704f9d442970002bacd332f222 Mon Sep 17 00:00:00 2001
From: Marty Oehme <marty.oehme@gmail.com>
Date: Wed, 14 Feb 2024 20:35:10 +0100
Subject: [PATCH] fix(code): Add logistic regression to regressions

---
 src/prep_data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/prep_data.py b/src/prep_data.py
index 619b58e..9c4af44 100644
--- a/src/prep_data.py
+++ b/src/prep_data.py
@@ -26,7 +26,7 @@ def calculate_validities(
 
     # needs to go lowest to highest in case of multiple mentioned approaches
     vd.loc[
-        vd[method_col].str.contains("|".join(["OLS", "ordinary.least.square"])),
+        vd[method_col].str.contains("|".join(["OLS", "ordinary.least.square", "logistic.regression"])),
         INT_COL_NAME,
     ] = 2.0
     vd.loc[