refactor: Adds tools.codemod.Ruff

#3630 (comment) **Introduces no new behavior**.
vega · Oct 16, 2024 · 0615228 · 0615228
1 parent 7cf627d
commit 0615228
Show file tree

Hide file tree

Showing 6 changed files with 287 additions and 153 deletions.
diff --git a/sphinxext/code_ref.py b/sphinxext/code_ref.py
@@ -10,7 +10,7 @@
 from sphinx.util.parsing import nested_parse_to_nodes
 
 from altair.vegalite.v5.schema._typing import VegaThemes
-from tools.codemod import embed_extract_func_def, extract_func_def
+from tools.codemod import extract_func_def, extract_func_def_embed
 
 if TYPE_CHECKING:
     import sys
@@ -222,11 +222,11 @@ def run(self) -> Sequence[nodes.Node]:
                 optgroup("Carbon", (option(nm) for nm in carbon_names)),
             )
         )
-        py_code = embed_extract_func_def(
+        py_code = extract_func_def_embed(
             module_name,
             func_name,
-            before_code=_before_code(REFRESH_NAME, SELECT_ID, TARGET_DIV_ID),
-            after_code=f"{REFRESH_NAME}()",
+            before=_before_code(REFRESH_NAME, SELECT_ID, TARGET_DIV_ID),
+            after=f"{REFRESH_NAME}()",
             assign_to="chart",
             indent=4,
         )