patched

jmafoster1 · jmafoster1 · commit 4918b2e4070b · 2024-03-18T13:12:56.000Z
diff --git a/causal_testing/testing/estimators.py b/causal_testing/testing/estimators.py
@@ -357,7 +357,7 @@ def estimate_coefficient(self) -> tuple[pd.Series, list[pd.Series, pd.Series]]:
                 self.df.dtypes[factor.name()] == "object"
                 for factor in patsy_md.rhs_termlist[1].factors
                 # We want to remove this long term as it prevents us from discovering categoricals within I(...) blocks
-                if factor in self.df
+                if factor.name() in self.df.dtypes
             )
         ):
             design_info = dmatrix(self.formula.split("~")[1], self.df).design_info

Original file line number	Diff line number	Diff line change
`@@ -357,7 +357,7 @@ def estimate_coefficient(self) -> tuple[pd.Series, list[pd.Series, pd.Series]]:`
`357`	`357`	`self.df.dtypes[factor.name()] == "object"`
`358`	`358`	`for factor in patsy_md.rhs_termlist[1].factors`
`359`	`359`	`# We want to remove this long term as it prevents us from discovering categoricals within I(...) blocks`
`360`		`- if factor in self.df`
	`360`	`+ if factor.name() in self.df.dtypes`
`361`	`361`	`)`
`362`	`362`	`):`
`363`	`363`	`design_info = dmatrix(self.formula.split("~")[1], self.df).design_info`