loosen pylint requirements

Author: Icíar Lloréns Jover

obsweatherscale/likelihoods/transformed_likelihood.py

@@ -100,7 +100,7 @@ def _shaped_noise_covar(
     def expected_log_prob(
         self,
         target: torch.Tensor,
-        input: MultivariateNormal,
+        input: MultivariateNormal, # pylint: disable=redefined-builtin
         *params: Any,
         **kwargs: Any,
     ) -> torch.Tensor:
@@ -138,6 +138,7 @@ def expected_log_prob(
         # Handle NaN values if enabled
         nan_policy = settings.observation_nan_policy.value()
         if nan_policy == "mask":
+            # pylint: disable=protected-access
             observed = settings.observation_nan_policy._get_observed(
                 target, input.event_shape
             )
@@ -338,6 +339,7 @@ def forward(
 
         # Remove NaN values if enabled
         if settings.observation_nan_policy.value() == "mask":
+            # pylint: disable=protected-access
             observed = settings.observation_nan_policy._get_observed(
                 target, output.event_shape
             )

obsweatherscale/models/gp_model.py

@@ -18,7 +18,7 @@ def __init__(
         likelihood: _GaussianLikelihoodBase,
         mean_module: Mean,
         covar_module: Kernel,
-    ) -> None:
+    ) -> None: # pylint: disable=arguments-differ
         """Initialize the GPModel.
 
         Parameters
@@ -38,6 +38,7 @@ def __init__(
         self.mean_module = mean_module
         self.covar_module = covar_module
 
+    # pylint: disable=arguments-differ
     def forward(self, x: torch.Tensor, **kwargs: Any) -> MultivariateNormal:
         mean_x = self.mean_module(x)
         covar_x = self.covar_module(x)

pyproject.toml

@@ -54,6 +54,13 @@ max-line-length = 120
 # Minimum number of public methods for a class (see R0903).
 min-public-methods = 0
 
+[tool.pylint.messages_control]
+disable = [
+    "too-many-arguments",
+    "too-many-positional-arguments",
+    "too-many-locals",
+]
+
 [tool.mypy]
 ignore_missing_imports = true
 disallow_untyped_defs = true