diff --git a/src/databricks/sql/utils.py b/src/databricks/sql/utils.py index 7ea0d7f5c..8682f0ad1 100644 --- a/src/databricks/sql/utils.py +++ b/src/databricks/sql/utils.py @@ -19,6 +19,7 @@ pyarrow = None from databricks.sql import OperationalError +from databricks.sql.exc import ProgrammingError from databricks.sql.cloudfetch.download_manager import ResultFileDownloadManager from databricks.sql.thrift_api.TCLIService.ttypes import ( TRowSet, @@ -548,7 +549,7 @@ def escape_args(self, parameters): elif isinstance(parameters, (list, tuple)): return tuple(self.escape_item(x) for x in parameters) else: - raise exc.ProgrammingError( + raise ProgrammingError( "Unsupported param format: {}".format(parameters) ) @@ -606,7 +607,7 @@ def escape_item(self, item): elif isinstance(item, Mapping): return self.escape_mapping(item) else: - raise exc.ProgrammingError("Unsupported object {}".format(item)) + raise ProgrammingError("Unsupported object {}".format(item)) def inject_parameters(operation: str, parameters: Dict[str, str]):