Merge pull request #34 from causy-dev/refactor/rename-pipeline-steps

refactor: rename pipeline steps

causy/causal_effect_estimation/multivariate_regression.py

@@ -18,7 +18,7 @@ class ComputeDirectEffectsMultivariateRegression(PipelineStepInterface):
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(self, nodes: Tuple[str], graph: BaseGraphInterface) -> TestResult:
+    def process(self, nodes: Tuple[str], graph: BaseGraphInterface) -> TestResult:
         """
         Calculate the direct effect of each edge in the graph using multivariate regression.
         :param nodes: list of nodes

causy/common_pipeline_steps/calculation.py

@@ -24,7 +24,7 @@ class CalculatePearsonCorrelations(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(self, nodes: Tuple[str], graph: BaseGraphInterface) -> TestResult:
+    def process(self, nodes: Tuple[str], graph: BaseGraphInterface) -> TestResult:
         """
         Calculate the correlation between each pair of nodes and store it to the respective edge.
         :param nodes: list of nodes

causy/common_pipeline_steps/placeholder.py

@@ -18,7 +18,7 @@ class PlaceholderTest(
     chunk_size_parallel_processing: int = 10
     parallel: bool = False
 
-    def test(
+    def process(
         self, nodes: Tuple[str], graph: BaseGraphInterface
     ) -> List[TestResult] | TestResult:
         """

causy/independence_tests/common.py

@@ -30,7 +30,9 @@ class CorrelationCoefficientTest(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(self, nodes: List[str], graph: BaseGraphInterface) -> Optional[TestResult]:
+    def process(
+        self, nodes: List[str], graph: BaseGraphInterface
+    ) -> Optional[TestResult]:
         """
         Test if u and v are independent and delete edge in graph if they are.
         :param nodes: list of nodes
@@ -65,7 +67,7 @@ class PartialCorrelationTest(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(
+    def process(
         self, nodes: Tuple[str], graph: BaseGraphInterface
     ) -> Optional[List[TestResult]]:
         """
@@ -142,7 +144,9 @@ class ExtendedPartialCorrelationTestMatrix(
     chunk_size_parallel_processing: int = 1000
     parallel: bool = False
 
-    def test(self, nodes: List[str], graph: BaseGraphInterface) -> Optional[TestResult]:
+    def process(
+        self, nodes: List[str], graph: BaseGraphInterface
+    ) -> Optional[TestResult]:
         """
         Test if nodes u,v are independent given Z (set of nodes) based on partial correlation using the inverted covariance matrix (precision matrix).
         https://en.wikipedia.org/wiki/Partial_correlation#Using_matrix_inversion
@@ -244,7 +248,9 @@ class ExtendedPartialCorrelationTestLinearRegression(
     chunk_size_parallel_processing: int = 1000
     parallel: bool = False
 
-    def test(self, nodes: List[str], graph: BaseGraphInterface) -> Optional[TestResult]:
+    def process(
+        self, nodes: List[str], graph: BaseGraphInterface
+    ) -> Optional[TestResult]:
         if not graph.edge_exists(graph.nodes[nodes[0]], graph.nodes[nodes[1]]):
             return
 

causy/interfaces.py

@@ -262,7 +262,9 @@ def name(self) -> str:
         return serialize_module_name(self)
 
     @abstractmethod
-    def test(self, nodes: List[str], graph: BaseGraphInterface) -> Optional[TestResult]:
+    def process(
+        self, nodes: List[str], graph: BaseGraphInterface
+    ) -> Optional[TestResult]:
         """
         Test if u and v are independent
         :param u: u values
@@ -274,7 +276,7 @@ def test(self, nodes: List[str], graph: BaseGraphInterface) -> Optional[TestResu
     def __call__(
         self, nodes: List[str], graph: BaseGraphInterface
     ) -> Optional[TestResult]:
-        return self.test(nodes, graph)
+        return self.process(nodes, graph)
 
 
 class ExitConditionInterface(ABC, BaseModel):

causy/orientation_rules/fci.py

@@ -21,7 +21,7 @@ class ColliderRuleFCI(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(
+    def process(
         self, nodes: Tuple[str], graph: BaseGraphInterface
     ) -> Optional[List[TestResult] | TestResult]:
         """

causy/orientation_rules/pc.py

@@ -27,7 +27,7 @@ class ColliderTest(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(
+    def process(
         self, nodes: Tuple[str], graph: BaseGraphInterface
     ) -> Optional[List[TestResult] | TestResult]:
         """
@@ -95,7 +95,7 @@ class NonColliderTest(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(
+    def process(
         self, nodes: Tuple[str], graph: BaseGraphInterface
     ) -> Optional[List[TestResult] | TestResult]:
         """
@@ -161,7 +161,7 @@ class FurtherOrientTripleTest(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(
+    def process(
         self, nodes: Tuple[str], graph: BaseGraphInterface
     ) -> Optional[List[TestResult] | TestResult]:
         """
@@ -219,7 +219,7 @@ class OrientQuadrupleTest(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(
+    def process(
         self, nodes: Tuple[str], graph: BaseGraphInterface
     ) -> Optional[List[TestResult] | TestResult]:
         """
@@ -287,7 +287,7 @@ class FurtherOrientQuadrupleTest(
     chunk_size_parallel_processing: int = 1
     parallel: bool = False
 
-    def test(
+    def process(
         self, nodes: Tuple[str], graph: BaseGraphInterface
     ) -> Optional[List[TestResult] | TestResult]:
         """