Skip to content

Commit 059657e

Browse files
add strong typing for manifest in _extract_description
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 82e0f8b commit 059657e

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

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

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import re
44
from typing import Dict, Tuple, List, Optional, Union, TYPE_CHECKING, Set
55

6+
from databricks.sql.backend.sea.models.base import ResultManifest
67
from databricks.sql.backend.sea.utils.constants import (
78
ALLOWED_SESSION_CONF_TO_DEFAULT_VALUES_MAP,
89
ResultFormat,
@@ -286,7 +287,9 @@ def get_allowed_session_configurations() -> List[str]:
286287
"""
287288
return list(ALLOWED_SESSION_CONF_TO_DEFAULT_VALUES_MAP.keys())
288289

289-
def _extract_description_from_manifest(self, manifest_obj) -> Optional[List]:
290+
def _extract_description_from_manifest(
291+
self, manifest: ResultManifest
292+
) -> Optional[List]:
290293
"""
291294
Extract column description from a manifest object.
292295
@@ -297,7 +300,7 @@ def _extract_description_from_manifest(self, manifest_obj) -> Optional[List]:
297300
Optional[List]: A list of column tuples or None if no columns are found
298301
"""
299302

300-
schema_data = manifest_obj.schema
303+
schema_data = manifest.schema
301304
columns_data = schema_data.get("columns", [])
302305

303306
if not columns_data:

0 commit comments

Comments
 (0)