fix

nbQA-dev · Nov 10, 2024 · fc2aecd · fc2aecd
1 parent 234638d
commit fc2aecd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/nbqa/handle_magics.py b/nbqa/handle_magics.py
@@ -29,8 +29,8 @@ def _get_node_args(func: ast.Call) -> List[str]:
     """
     args = []
     for arg in func.args:
-        if isinstance(arg, ast.Str):
-            args.append(arg.s)
+        if isinstance(arg, ast.Constant):
+            args.append(arg.value)
         else:
             raise AssertionError(
                 "Please report a bug at https://github.com/nbQA-dev/nbQA/issues"
@@ -158,8 +158,8 @@ def visit_Call(self, node: ast.Call) -> None:
             if node.func.attr == "run_cell_magic":
                 args = []
                 for arg in node.args:
-                    if isinstance(arg, ast.Str):
-                        args.append(arg.s)
+                    if isinstance(arg, ast.Constant):
+                        args.append(arg.value)
                     else:
                         raise AssertionError(
                             "Please report a bug at https://github.com/nbQA-dev/nbQA/issues"