Skip to content

Commit a58aaae

Browse files
committed
refactor : change get_table_lineage return type to List[str, dict]
1 parent e1a700d commit a58aaae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

data_utils/datahub_source.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def get_table_lineage(
8888
degree_values (List[str]): Degree filter values like ["1", "2", "3+"]. Defaults to ["1", "2"].
8989
9090
Returns:
91-
List[Tuple[str, dict]]: A list containing the dataset URN and its lineage result.
91+
List[str, dict]: A list containing the dataset URN and its lineage result.
9292
"""
9393

9494
if degree_values is None:
@@ -133,7 +133,7 @@ def get_table_lineage(
133133
}
134134

135135
result = graph.execute_graphql(query=query, variables=variables)
136-
return [(urn, result)]
136+
return urn, result
137137

138138
def get_column_lineage(self, urn):
139139
# URN에 대한 UPSTREAM lineage의 column source를 가져오는 함수

0 commit comments

Comments
 (0)