format

Signed-off-by: Nicholas Parente <[email protected]>
py-why · Jan 28, 2025 · 710b045 · 710b045
1 parent 310434d
commit 710b045
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/dowhy/datasets.py b/dowhy/datasets.py
@@ -846,7 +846,9 @@ def dataset_from_random_graph(
         else:
             variable_type_dict[idx] = BINARY
 
-    return linear_dataset_from_graph(DAG, treatments, outcome, variable_type_dict=variable_type_dict, num_samples=num_samples)
+    return linear_dataset_from_graph(
+        DAG, treatments, outcome, variable_type_dict=variable_type_dict, num_samples=num_samples
+    )
 
 
 def partially_linear_dataset(

diff --git a/tests/causal_estimators/test_propensity_score_matching_estimator.py b/tests/causal_estimators/test_propensity_score_matching_estimator.py
@@ -17,7 +17,7 @@ class TestPropensityScoreMatchingEstimator(object):
             "num_treatments",
             "treatment_is_binary",
             "outcome_is_binary",
-            "identifier_method"
+            "identifier_method",
         ],
         [
             (
@@ -37,7 +37,7 @@ class TestPropensityScoreMatchingEstimator(object):
                 [
                     False,
                 ],
-                "backdoor"
+                "backdoor",
             ),
             (
                 0.3,
@@ -56,7 +56,7 @@ class TestPropensityScoreMatchingEstimator(object):
                 [
                     False,
                 ],
-                "general_adjustment"
+                "general_adjustment",
             ),
         ],
     )
@@ -70,7 +70,7 @@ def test_average_treatment_effect(
         num_treatments,
         treatment_is_binary,
         outcome_is_binary,
-        identifier_method
+        identifier_method,
     ):
         estimator_tester = SimpleEstimator(error_tolerance, Estimator, identifier_method=identifier_method)
         estimator_tester.average_treatment_effect_testsuite(