Skip to content

Commit 93e2f85

Browse files
small fix to llm extraction file (#162)
1 parent 31b9a8b commit 93e2f85

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

eureka_ml_insights/user_configs/llm_extraction.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def configure_subpipeline(
149149
# consolidate new {extracted_answer_col} to replace the original {extracted_answer_col} whenever the latter is empty
150150
# the initial if statement checks whether there has been a join beforehand
151151
"transform": RunPythonTransform(
152-
f"df['{extracted_answer_col}'] = df.apply(lambda row: row['{extracted_answer_col}'] if '{extracted_answer_col}_x' not in row else row['{extracted_answer_col}_y'] if row['{extracted_answer_col}_x'] == '' else row['{extracted_answer_col}_x'], axis=1)"
152+
f"df['{extracted_answer_col}'] = df.apply(lambda row: row['{extracted_answer_col}'] if '{extracted_answer_col}_x' not in row else row['{extracted_answer_col}_y'] if row['{extracted_answer_col}_x'] == '{not_extracted_answer_value}' else row['{extracted_answer_col}_x'], axis=1)"
153153
),
154154
},
155155
),

0 commit comments

Comments
 (0)