diff --git a/src/sql/command.py b/src/sql/command.py index c0febe58a..8dc953508 100644 --- a/src/sql/command.py +++ b/src/sql/command.py @@ -90,7 +90,7 @@ def __init__(self, magic, user_ns, line, cell, connection=None) -> None: self.parsed["connection"] = self.args.line[0] if connection is not None: - self.parsed['connection'] = connection + self.parsed["connection"] = connection if self.args.with_: self.args.with_ = [ diff --git a/src/sql/magic.py b/src/sql/magic.py index 6711c679f..7a5f6e8b4 100644 --- a/src/sql/magic.py +++ b/src/sql/magic.py @@ -363,11 +363,17 @@ def execute(self, line="", cell="", local_ns=None, connection=None): """ return self._execute( - line=line, cell=cell, local_ns=local_ns, is_interactive_mode=False, connection=connection + line=line, + cell=cell, + local_ns=local_ns, + is_interactive_mode=False, + connection=connection, ) @modify_exceptions - def _execute(self, line, cell, local_ns, is_interactive_mode=False, connection=None): + def _execute( + self, line, cell, local_ns, is_interactive_mode=False, connection=None + ): """ This function implements the cell logic; we create this private method so we can control how the function is called. Otherwise,