new util in plot

ploomber · Jan 21, 2024 · ca09ea4 · ca09ea4
1 parent 57dfccf
commit ca09ea4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/sql/magic_plot.py b/src/sql/magic_plot.py
@@ -86,9 +86,7 @@ def execute(self, line="", cell="", local_ns=None):
 
         cmd = SQLPlotCommand(self, line)
 
-        util.expand_args(
-            cmd.args, user_ns, ["table", "column", "schema", "with_", "orient"]
-        )
+        util.expand_args_cmd(cmd.args, user_ns)
 
         if len(cmd.args.column) == 1:
             column = cmd.args.column[0]