Fix case with parameter after *args (#14)

amazon-braket · Aug 23, 2023 · 619e9b9 · 619e9b9
1 parent 9918625
commit 619e9b9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/braket/flake8_plugins/braket_checkstyle_plugin.py b/src/braket/flake8_plugins/braket_checkstyle_plugin.py
@@ -259,7 +259,7 @@ def _check_argument_order(
     ) -> None:
         expected_index = 0
         if context.previous_arg is None:
-            if node.args.args[0].arg in self.RESERVED_ARGS:
+            if node.args.args and node.args.args[0].arg in self.RESERVED_ARGS:
                 expected_index = 1
         elif arg_type == context.previous_arg[1]:
             expected_index = context.previous_arg[0] + 1

diff --git a/test/unit_tests/braket/flake8_plugins/example_files/no_error_functions.py b/test/unit_tests/braket/flake8_plugins/example_files/no_error_functions.py
@@ -151,6 +151,16 @@ def function_14(name: str, *, control: int = 0, target: int = 1) -> int:
     pass
 
 
+def function_15(*args, a0: int) -> int:
+    """This is a description.
+    Args:
+        a0 (int): This is a parameter
+    Returns:
+        int: value of the function
+    """
+    pass
+
+
 class MyClass:
     def __init__(self, a0:int):
         """