Skip to content

Commit a25f22b

Browse files
author
nik
committed
Fix checking _adala_errors in ner_extraction
1 parent a5e3a24 commit a25f22b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

adala/skills/collection/entity_extraction.py

+3
Original file line numberDiff line numberDiff line change
@@ -267,6 +267,9 @@ def extract_indices(self, df):
267267
input_field_name = self._get_input_field_name()
268268
output_field_name = self._get_output_field_name()
269269
for i, row in df.iterrows():
270+
if row.get('_adala_error'):
271+
logger.warning(f"Error in row {i}: {row['_adala_message']}")
272+
continue
270273
text = row[input_field_name]
271274
entities = row[output_field_name]
272275
to_remove = []

0 commit comments

Comments
 (0)