Skip to content

Commit b00c06c

Browse files
reduce diff
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 9e4ef2e commit b00c06c

File tree

4 files changed

+5
-5
lines changed

4 files changed

+5
-5
lines changed

examples/experimental/tests/test_sea_async_query.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from databricks.sql.client import Connection
99
from databricks.sql.backend.types import CommandState
1010

11-
logging.basicConfig(level=logging.DEBUG)
11+
logging.basicConfig(level=logging.INFO)
1212
logger = logging.getLogger(__name__)
1313

1414

examples/experimental/tests/test_sea_sync_query.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import logging
77
from databricks.sql.client import Connection
88

9-
logging.basicConfig(level=logging.DEBUG)
9+
logging.basicConfig(level=logging.INFO)
1010
logger = logging.getLogger(__name__)
1111

1212

src/databricks/sql/backend/databricks_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def execute_command(
9191
lz4_compression: bool,
9292
cursor: "Cursor",
9393
use_cloud_fetch: bool,
94-
parameters: List[ttypes.TSparkParameter],
94+
parameters: List,
9595
async_op: bool,
9696
enforce_embedded_schema_correctness: bool,
9797
) -> Union["ResultSet", None]:

src/databricks/sql/backend/sea/backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -556,11 +556,11 @@ def get_query_state(self, command_id: CommandId) -> CommandState:
556556
CommandState: The current state of the command
557557
558558
Raises:
559-
ProgrammingError: If the command ID is invalid
559+
ValueError: If the command ID is invalid
560560
"""
561561

562562
if command_id.backend_type != BackendType.SEA:
563-
raise ProgrammingError("Not a valid SEA command ID")
563+
raise ValueError("Not a valid SEA command ID")
564564

565565
sea_statement_id = command_id.to_sea_statement_id()
566566
if sea_statement_id is None:

0 commit comments

Comments
 (0)