Skip to content

Commit 17a65db

Browse files
align description with Thrift
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent c19732f commit 17a65db

File tree

1 file changed

+13
-5
lines changed

1 file changed

+13
-5
lines changed

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

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -318,15 +318,23 @@ def _extract_description_from_manifest(
318318
columns = []
319319
for col_data in columns_data:
320320
# Format: (name, type_code, display_size, internal_size, precision, scale, null_ok)
321+
name = col_data.get("name", "")
322+
type_name = col_data.get("type_name", "")
323+
type_name = (
324+
type_name[:-5] if type_name.endswith("_TYPE") else type_name
325+
).lower()
326+
precision = col_data.get("type_precision")
327+
scale = col_data.get("type_scale")
328+
321329
columns.append(
322330
(
323-
col_data.get("name", ""), # name
324-
col_data.get("type_name", ""), # type_code
331+
name, # name
332+
type_name, # type_code
325333
None, # display_size (not provided by SEA)
326334
None, # internal_size (not provided by SEA)
327-
col_data.get("precision"), # precision
328-
col_data.get("scale"), # scale
329-
col_data.get("nullable", True), # null_ok
335+
precision, # precision
336+
scale, # scale
337+
None, # null_ok
330338
)
331339
)
332340

0 commit comments

Comments
 (0)