apply ner bugfix to legacy skill too

HumanSignal · Nov 6, 2024 · f40c449 · f40c449
1 parent 01c8b8e
commit f40c449
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/adala/skills/collection/entity_extraction.py b/adala/skills/collection/entity_extraction.py
@@ -354,6 +354,7 @@ def extract_indices(self, df):
         """
         input_field_name = self._get_input_field_name()
         output_field_name = self._get_output_field_name()
+        df = validate_output_format_for_ner_tag(df, input_field_name, output_field_name)
         df = extract_indices(df, input_field_name, output_field_name, self._quote_string_field_name, self._labels_field_name)
         return df