Fix: supress lintings locally

Author: f-allian

causal_testing/surrogate/causal_surrogate_assisted.py

@@ -1,5 +1,3 @@
-# pylint: disable=too-few-public-methods
-
 """Module containing classes to define and run causal surrogate assisted test cases"""
 
 from abc import ABC, abstractmethod
@@ -21,7 +19,7 @@ class SimulationResult:
     relationship: str
 
 
-class SearchAlgorithm(ABC):
+class SearchAlgorithm(ABC): # pylint: disable=too-few-public-methods
     """Class to be inherited with the search algorithm consisting of a search function and the fitness function of the
     space to be searched"""
 

causal_testing/surrogate/surrogate_search_algorithms.py

@@ -1,6 +1,4 @@
 """Module containing implementation of search algorithm for surrogate search """
-# pylint: disable=cell-var-from-loop
-# pylint: disable=too-many-locals
 # Fitness functions are required to be iteratively defined, including all variables within.
 
 from operator import itemgetter
@@ -26,6 +24,7 @@ def __init__(self, delta=0.05, config: dict = None) -> None:
             "some_effect": lambda x: abs(1 / x),
         }
 
+    # pylint: disable=too-many-locals
     def search(
             self, surrogate_models: list[CubicSplineRegressionEstimator], specification: CausalSpecification
     ) -> list:
@@ -36,6 +35,7 @@ def search(
 
             # The GA fitness function after including required variables into the function's scope
             # Unused arguments are required for pygad's fitness function signature
+            #pylint: disable=cell-var-from-loop
             def fitness_function(ga, solution, idx): # pylint: disable=unused-argument
                 surrogate.control_value = solution[0] - self.delta
                 surrogate.treatment_value = solution[0] + self.delta