diff --git a/causy/causal_discovery/constraint/algorithms/pc.py b/causy/causal_discovery/constraint/algorithms/pc.py index 83075b8..ffe60b2 100644 --- a/causy/causal_discovery/constraint/algorithms/pc.py +++ b/causy/causal_discovery/constraint/algorithms/pc.py @@ -91,9 +91,7 @@ PCClassic = graph_model_factory( Algorithm( pipeline_steps=[ - CalculatePearsonCorrelations( - display_name="Calculate Pearson Correlations" - ), + CalculatePearsonCorrelations(display_name="Calculate Pearson Correlations"), CorrelationCoefficientTest( threshold=VariableReference(name="threshold"), display_name="Correlation Coefficient Test", diff --git a/causy/causal_discovery/constraint/independence_tests/common.py b/causy/causal_discovery/constraint/independence_tests/common.py index 6fbea72..aa3f84b 100644 --- a/causy/causal_discovery/constraint/independence_tests/common.py +++ b/causy/causal_discovery/constraint/independence_tests/common.py @@ -256,7 +256,9 @@ def process( "triple": [ graph.nodes[nodes[0]], graph.nodes[nodes[1]], - nodes_set - {graph.nodes[nodes[0]], graph.nodes[nodes[1]]}, + list( + nodes_set - {graph.nodes[nodes[0]], graph.nodes[nodes[1]]} + ), ], }, ) @@ -269,7 +271,9 @@ def process( "triple": [ graph.nodes[nodes[0]], graph.nodes[nodes[1]], - nodes_set - {graph.nodes[nodes[0]], graph.nodes[nodes[1]]}, + list( + nodes_set - {graph.nodes[nodes[0]], graph.nodes[nodes[1]]} + ), ], }, ) @@ -349,7 +353,9 @@ def process( "triple": [ graph.nodes[nodes[0]], graph.nodes[nodes[1]], - nodes_set - {graph.nodes[nodes[0]], graph.nodes[nodes[1]]}, + list( + nodes_set - {graph.nodes[nodes[0]], graph.nodes[nodes[1]]} + ), ], }, ) @@ -362,7 +368,9 @@ def process( "triple": [ graph.nodes[nodes[0]], graph.nodes[nodes[1]], - nodes_set - {graph.nodes[nodes[0]], graph.nodes[nodes[1]]}, + list( + nodes_set - {graph.nodes[nodes[0]], graph.nodes[nodes[1]]} + ), ] }, ) diff --git a/tests/test_pc_e2e.py b/tests/test_pc_e2e.py index 009890b..32a2da8 100644 --- a/tests/test_pc_e2e.py +++ b/tests/test_pc_e2e.py @@ -3,7 +3,8 @@ PC, PC_ORIENTATION_RULES, PC_GRAPH_UI_EXTENSION, - PC_DEFAULT_THRESHOLD, PCClassic, + PC_DEFAULT_THRESHOLD, + PCClassic, ) from causy.causal_effect_estimation.multivariate_regression import ( ComputeDirectEffectsInDAGsMultivariateRegression,