From 2c4734867cf87b5cf578cbf85a8cc4eaa8cfa802 Mon Sep 17 00:00:00 2001 From: Amit Sharma Date: Sun, 14 Apr 2024 16:36:52 +0530 Subject: [PATCH] updated formatting Signed-off-by: Amit Sharma --- dowhy/causal_estimators/distance_matching_estimator.py | 9 ++------- dowhy/causal_refuters/add_unobserved_common_cause.py | 8 ++++---- dowhy/causal_refuters/bootstrap_refuter.py | 2 +- dowhy/causal_refuters/data_subset_refuter.py | 2 +- dowhy/causal_refuters/dummy_outcome_refuter.py | 4 ++-- dowhy/causal_refuters/evalue_sensitivity_analyzer.py | 2 +- dowhy/causal_refuters/placebo_treatment_refuter.py | 2 +- dowhy/causal_refuters/random_common_cause.py | 2 +- 8 files changed, 13 insertions(+), 18 deletions(-) diff --git a/dowhy/causal_estimators/distance_matching_estimator.py b/dowhy/causal_estimators/distance_matching_estimator.py index b947ffdb5f..7b74bfc4df 100644 --- a/dowhy/causal_estimators/distance_matching_estimator.py +++ b/dowhy/causal_estimators/distance_matching_estimator.py @@ -103,12 +103,7 @@ def __init__( self.matched_indices_att = None self.matched_indices_atc = None - def fit( - self, - data: pd.DataFrame, - effect_modifier_names: Optional[List[str]] = None, - exact_match_cols=None - ): + def fit(self, data: pd.DataFrame, effect_modifier_names: Optional[List[str]] = None, exact_match_cols=None): """ Fits the estimator with data for effect estimation :param data: data frame containing the data @@ -121,7 +116,7 @@ def fit( methods support this currently. """ self.exact_match_cols = exact_match_cols - self._fit_params = {'exact_match_cols': exact_match_cols} + self._fit_params = {"exact_match_cols": exact_match_cols} self.reset_encoders() # Forget any existing encoders self._set_effect_modifiers(data, effect_modifier_names) diff --git a/dowhy/causal_refuters/add_unobserved_common_cause.py b/dowhy/causal_refuters/add_unobserved_common_cause.py index fbc9922962..e6ce8fe887 100755 --- a/dowhy/causal_refuters/add_unobserved_common_cause.py +++ b/dowhy/causal_refuters/add_unobserved_common_cause.py @@ -843,7 +843,7 @@ def sensitivity_simulation( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, @@ -891,7 +891,7 @@ def sensitivity_simulation( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, @@ -967,7 +967,7 @@ def sensitivity_simulation( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, @@ -1025,7 +1025,7 @@ def sensitivity_simulation( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, diff --git a/dowhy/causal_refuters/bootstrap_refuter.py b/dowhy/causal_refuters/bootstrap_refuter.py index d12ae2e8e2..6a340a1e8a 100644 --- a/dowhy/causal_refuters/bootstrap_refuter.py +++ b/dowhy/causal_refuters/bootstrap_refuter.py @@ -133,7 +133,7 @@ def _refute_once( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, diff --git a/dowhy/causal_refuters/data_subset_refuter.py b/dowhy/causal_refuters/data_subset_refuter.py index 92b6066cb6..5b6a2b3f9f 100755 --- a/dowhy/causal_refuters/data_subset_refuter.py +++ b/dowhy/causal_refuters/data_subset_refuter.py @@ -77,7 +77,7 @@ def _refute_once( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, diff --git a/dowhy/causal_refuters/dummy_outcome_refuter.py b/dowhy/causal_refuters/dummy_outcome_refuter.py index 5c9ff660c7..7ca55e23eb 100644 --- a/dowhy/causal_refuters/dummy_outcome_refuter.py +++ b/dowhy/causal_refuters/dummy_outcome_refuter.py @@ -494,7 +494,7 @@ def refute_dummy_outcome( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, @@ -574,7 +574,7 @@ def refute_dummy_outcome( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, diff --git a/dowhy/causal_refuters/evalue_sensitivity_analyzer.py b/dowhy/causal_refuters/evalue_sensitivity_analyzer.py index e56fc7403e..a883047778 100644 --- a/dowhy/causal_refuters/evalue_sensitivity_analyzer.py +++ b/dowhy/causal_refuters/evalue_sensitivity_analyzer.py @@ -262,7 +262,7 @@ def benchmark(self, data: pd.DataFrame): new_estimator.fit( self.data, effect_modifier_names=self.estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) # new effect estimate diff --git a/dowhy/causal_refuters/placebo_treatment_refuter.py b/dowhy/causal_refuters/placebo_treatment_refuter.py index 3d51e52b62..546b412100 100755 --- a/dowhy/causal_refuters/placebo_treatment_refuter.py +++ b/dowhy/causal_refuters/placebo_treatment_refuter.py @@ -155,7 +155,7 @@ def _refute_once( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data, diff --git a/dowhy/causal_refuters/random_common_cause.py b/dowhy/causal_refuters/random_common_cause.py index 41fe316f25..98b54a720a 100755 --- a/dowhy/causal_refuters/random_common_cause.py +++ b/dowhy/causal_refuters/random_common_cause.py @@ -71,7 +71,7 @@ def _refute_once( new_estimator.fit( new_data, effect_modifier_names=estimate.estimator._effect_modifier_names, - **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {} + **new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}, ) new_effect = new_estimator.estimate_effect( new_data,