Removed dead code

Author: jmafoster1

causal_testing/__main__.py

@@ -216,14 +216,13 @@ def main() -> None:
             # Need to reset index to allow for multiple files having the same index (i.e. starting at zero).
             # Otherwise you end up with duplicate indices, which causes problems further down the line
             df = pd.concat([read_dataframe(path) for path in args.data_paths]).reset_index()
-            print(df)
             if args.variables:
                 df = df[args.variables]
             # Drop unnamed columns
             unnamed_columns = [c for c in df.columns if c.startswith("Unnamed: ")]
             if unnamed_columns:
                 warn(f"Dropping unnamed columns: {unnamed_columns}")
-            df = df.drop(unnamed_columns)
+            df = df.drop(unnamed_columns, axis=1)
 
             discover_class = discover_map[args.technique].load()
             discover = discover_class(

causal_testing/estimation/abstract_regression_estimator.py

@@ -182,9 +182,6 @@ def _predict(self, df) -> pd.DataFrame:
         for k, v in self.adjustment_config.items():
             x[k] = v
         x = dmatrix(self.formula.split("~")[1], x, return_type="dataframe")
-        for col in x:
-            if isinstance(x[col], pd.CategoricalDtype) or pd.api.types.is_object_dtype(x[col]):
-                x = pd.get_dummies(x, columns=[col], drop_first=True)
 
         return model.get_prediction(x).summary_frame()
 

tests/estimation_tests/test_linear_regression_estimator.py

@@ -146,7 +146,7 @@ def test_no_formula_of_adjustment_set(self):
         """
         with self.assertRaises(ValueError) as e:
             LinearRegressionEstimator(treatment_variable="cut", outcome_variable="price")
-            self.assertEqual(e.exception, f"Please specify either a formula or an adjustment set.")
+            self.assertEqual(e.exception, "Please specify either a formula or an adjustment set.")
 
     def test_linear_regression_categorical_ate(self):
         df = self.scarf_df.copy()