def _extract_domain_entities_rltn_score()

in src/entity_relation_scorer.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

def _extract_domain_entities_rltn_score(df, entity_name, relation, tgt_entity_name, score_col=None):
    sel_df = df.loc[~df[entity_name].isna(), [entity_name, tgt_entity_name, score_col]].reset_index(drop=True)
    for _, row in sel_df.iterrows():
        yield _generate_entity_rltn_score(row[entity_name], entity_name, relation, row[tgt_entity_name], row[score_col])