File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed
Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -168,7 +168,10 @@ def get_column_lineage(self, urn):
168168 # upstream_dataset별로 column lineage
169169 upstream_map = defaultdict (list )
170170
171- for fg in result .fineGrainedLineages :
171+ for fg in result .fineGrainedLineages or []:
172+ confidence_score = (
173+ fg .confidenceScore if fg .confidenceScore is not None else 1.0
174+ )
172175 for down in fg .downstreams :
173176 down_column = down .split ("," )[- 1 ].replace (")" , "" )
174177 for up in fg .upstreams :
@@ -177,7 +180,11 @@ def get_column_lineage(self, urn):
177180 up_column = up .split ("," )[- 1 ].replace (")" , "" )
178181
179182 upstream_map [up_dataset ].append (
180- {"upstream_column" : up_column , "downstream_column" : down_column }
183+ {
184+ "upstream_column" : up_column ,
185+ "downstream_column" : down_column ,
186+ "confidence" : confidence_score ,
187+ }
181188 )
182189
183190 # 최종 결과 구조 생성
You can’t perform that action at this time.
0 commit comments