Skip to content

Commit 02b3ae6

Browse files
amit-sharmaamit-sharma
committed
updated formatting
Signed-off-by: Amit Sharma <amit_sharma@live.com>
1 parent bc7de74 commit 02b3ae6

8 files changed

+13
-18
lines changed

‎dowhy/causal_estimators/distance_matching_estimator.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -103,12 +103,7 @@ def __init__(
103103
self.matched_indices_att = None
104104
self.matched_indices_atc = None
105105

106-
def fit(
107-
self,
108-
data: pd.DataFrame,
109-
effect_modifier_names: Optional[List[str]] = None,
110-
exact_match_cols=None
111-
):
106+
def fit(self, data: pd.DataFrame, effect_modifier_names: Optional[List[str]] = None, exact_match_cols=None):
112107
"""
113108
Fits the estimator with data for effect estimation
114109
:param data: data frame containing the data
@@ -121,7 +116,7 @@ def fit(
121116
methods support this currently.
122117
"""
123118
self.exact_match_cols = exact_match_cols
124-
self._fit_params = {'exact_match_cols': exact_match_cols}
119+
self._fit_params = {"exact_match_cols": exact_match_cols}
125120
self.reset_encoders() # Forget any existing encoders
126121
self._set_effect_modifiers(data, effect_modifier_names)
127122

‎dowhy/causal_refuters/add_unobserved_common_cause.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -843,7 +843,7 @@ def sensitivity_simulation(
843843
new_estimator.fit(
844844
new_data,
845845
effect_modifier_names=estimate.estimator._effect_modifier_names,
846-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
846+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
847847
)
848848
new_effect = new_estimator.estimate_effect(
849849
new_data,
@@ -891,7 +891,7 @@ def sensitivity_simulation(
891891
new_estimator.fit(
892892
new_data,
893893
effect_modifier_names=estimate.estimator._effect_modifier_names,
894-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
894+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
895895
)
896896
new_effect = new_estimator.estimate_effect(
897897
new_data,
@@ -967,7 +967,7 @@ def sensitivity_simulation(
967967
new_estimator.fit(
968968
new_data,
969969
effect_modifier_names=estimate.estimator._effect_modifier_names,
970-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
970+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
971971
)
972972
new_effect = new_estimator.estimate_effect(
973973
new_data,
@@ -1025,7 +1025,7 @@ def sensitivity_simulation(
10251025
new_estimator.fit(
10261026
new_data,
10271027
effect_modifier_names=estimate.estimator._effect_modifier_names,
1028-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
1028+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
10291029
)
10301030
new_effect = new_estimator.estimate_effect(
10311031
new_data,

‎dowhy/causal_refuters/bootstrap_refuter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def _refute_once(
133133
new_estimator.fit(
134134
new_data,
135135
effect_modifier_names=estimate.estimator._effect_modifier_names,
136-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
136+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
137137
)
138138
new_effect = new_estimator.estimate_effect(
139139
new_data,

‎dowhy/causal_refuters/data_subset_refuter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def _refute_once(
7777
new_estimator.fit(
7878
new_data,
7979
effect_modifier_names=estimate.estimator._effect_modifier_names,
80-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
80+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
8181
)
8282
new_effect = new_estimator.estimate_effect(
8383
new_data,

‎dowhy/causal_refuters/dummy_outcome_refuter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -494,7 +494,7 @@ def refute_dummy_outcome(
494494
new_estimator.fit(
495495
new_data,
496496
effect_modifier_names=estimate.estimator._effect_modifier_names,
497-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
497+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
498498
)
499499
new_effect = new_estimator.estimate_effect(
500500
new_data,
@@ -574,7 +574,7 @@ def refute_dummy_outcome(
574574
new_estimator.fit(
575575
new_data,
576576
effect_modifier_names=estimate.estimator._effect_modifier_names,
577-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
577+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
578578
)
579579
new_effect = new_estimator.estimate_effect(
580580
new_data,

‎dowhy/causal_refuters/evalue_sensitivity_analyzer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ def benchmark(self, data: pd.DataFrame):
262262
new_estimator.fit(
263263
self.data,
264264
effect_modifier_names=self.estimate.estimator._effect_modifier_names,
265-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
265+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
266266
)
267267

268268
# new effect estimate

‎dowhy/causal_refuters/placebo_treatment_refuter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def _refute_once(
155155
new_estimator.fit(
156156
new_data,
157157
effect_modifier_names=estimate.estimator._effect_modifier_names,
158-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
158+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
159159
)
160160
new_effect = new_estimator.estimate_effect(
161161
new_data,

‎dowhy/causal_refuters/random_common_cause.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def _refute_once(
7171
new_estimator.fit(
7272
new_data,
7373
effect_modifier_names=estimate.estimator._effect_modifier_names,
74-
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {}
74+
**new_estimator._fit_params if hasattr(new_estimator, "_fit_params") else {},
7575
)
7676
new_effect = new_estimator.estimate_effect(
7777
new_data,

0 commit comments

Comments
 (0)