exclude self

PaulinaPacyna · Nov 3, 2023 · 70d09f2 · 70d09f2
1 parent 2bbc97d
commit 70d09f2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/annotation_checker/checkers.py b/annotation_checker/checkers.py
@@ -20,7 +20,7 @@ class Checker(ABC):
     def __init__(
         self,
         exclude_parameters: str = "",
-        exclude_self: bool = True,
+        exclude_self: bool = False,
         exclude_by_name: str = "",
     ) -> None:
         self._errors = []
@@ -80,7 +80,7 @@ class FunctionChecker(Checker):
     def __init__(
         self,
         exclude_parameters: str = "",
-        exclude_self: bool = True,
+        exclude_self: bool = False,
         exclude_by_name: str = "",
     ) -> None:
         super().__init__(
@@ -110,10 +110,10 @@ def __check_args(self, function: ast.FunctionDef) -> None:
         ----------
             function (ast.FunctionDef): the function to be checked
         """
-        args = function.args
+        args = function.args.args
         if self._exclude_self:
             args = args[1:]
-        for argument in args.args:
+        for argument in args:
             if not argument.annotation:
                 if self.__check_if_not_excluded(argument.arg):
                     self._errors.append(

diff --git a/annotation_checker/main.py b/annotation_checker/main.py
@@ -44,7 +44,7 @@ def check_annotated(
             exclusion_comment=exclusion_comment,
         )
         function_checker = FunctionChecker(
-            exclude_parameters=exclude_parameters, exclude_self=exclude_self
+            exclude_parameters=exclude_parameters, exclude_self=False
         )
         class_checker = ClassChecker(
             exclude_parameters=exclude_parameters, exclude_self=exclude_self