pylint

Author: jmafoster1

causal_testing/estimation/gp.py

@@ -75,12 +75,20 @@ def mut_insert(expression: gp.PrimitiveTree, pset: gp.PrimitiveSet):
     return (expression,)
 
 
-def create_power_function(power):
+def create_power_function(order: int):
+    """
+    Creates a power operator and its corresponding sympy conversion.
+
+    :param order: The order of the power, e.g. `order=2` will give x^2.
+
+    :return: A pair consisting of the power function and the sympy conversion
+    """
+
     def power_func(x):
-        return power(x, power)
+        return power(x, order)
 
-    def sympy_conversion(x1):
-        return f"Pow({x1},{i})"
+    def sympy_conversion(x):
+        return f"Pow({x},{order})"
 
     return power_func, sympy_conversion
 
@@ -100,7 +108,8 @@ def __init__(
         sympy_conversions: dict = None,
         seed=0,
     ):
-        # pylint: disable=too-many-arguments,too-many-instance-attributes
+        # pylint: disable=too-many-arguments
+        # pylint: disable=too-many-instance-attributes
         random.seed(seed)
         self.df = df
         self.features = features

causal_testing/estimation/linear_regression_estimator.py

@@ -78,9 +78,10 @@ def gp_formula(
         :param ngen: The maximum number of GP generations to run for.
         :param pop_size: The GP population size.
         :param num_offspring: The number of offspring per generation.
-        :param max_order: The maximum polynomial order to use, e.g. `max_order=2` will give polynomials of the form `ax^2 + bx + c`.
-        :param extra_operators: Additional operators for the GP (defaults are +, *, log(x), and 1/x). Operations should be of
-                                the form (fun, numArgs), e.g. (add, 2).
+        :param max_order: The maximum polynomial order to use, e.g. `max_order=2` will give
+                          polynomials of the form `ax^2 + bx + c`.
+        :param extra_operators: Additional operators for the GP (defaults are +, *, log(x), and 1/x).
+                                Operations should be of the form (fun, numArgs), e.g. (add, 2).
         :param sympy_conversions: Dictionary of conversions of extra_operators for sympy,
                                   e.g. `"mul": lambda *args_: "Mul({},{})".format(*args_)`.
         :param seeds: Seed individuals for the population (e.g. if you think that the relationship between X and Y is