import

ploomber · Dec 31, 2023 · edf2efc · edf2efc
1 parent a1988c5
commit edf2efc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/sql/magic.py b/src/sql/magic.py
@@ -40,7 +40,6 @@
 from sql.magic_cmd import SqlCmdMagic
 from sql._patch import patch_ipython_usage_error
 from sql import util
-from sql.util import pretty_print, expand_args
 from sql.error_handler import handle_exception
 from sql._current import _set_sql_magic
 
@@ -410,7 +409,7 @@ def interactive_execute_wrapper(**kwargs):
 
         args = command.args
 
-        expand_args(args, user_ns, ["section", "close", "alias", "save"])
+        util.expand_args(args, user_ns, ["section", "close", "alias", "save"])
 
         if args.section and args.alias:
             raise exceptions.UsageError(
@@ -438,7 +437,7 @@ def interactive_execute_wrapper(**kwargs):
                         command.set_sql_with(with_)
                         display.message(
                             f"Generating CTE with stored snippets: \
-{pretty_print(with_)}"
+{util.pretty_print(with_)}"
                         )
                 else:
                     with_ = None