fix tests

HumanSignal · Nov 7, 2024 · fedb845 · fedb845
1 parent f40c449
commit fedb845
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/adala/skills/collection/entity_extraction.py b/adala/skills/collection/entity_extraction.py
@@ -354,7 +354,6 @@ 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
 

diff --git a/tests/test_label_studio_skill.py b/tests/test_label_studio_skill.py
@@ -392,7 +392,7 @@ async def test_label_studio_skill_valid_predictions():
                         f"Failed {n_failed_preds} predictions for {label_config} {model}"
                     )
                     predictions = predictions[is_success]
-
+            
             # filter out adala fields and input field
             predictions = predictions[
                 [
@@ -417,5 +417,4 @@ async def test_label_studio_skill_valid_predictions():
                     errored_configs.append((label_config, model, prediction, e))
 
     assert len(failed_configs) == 0, f"Failed configs: {failed_configs}"
-    # TODO DIA-1600: intermittent nans are sneaking through for Labels
-    # assert len(errored_configs) == 0, f"Errored configs: {errored_configs}"
+    assert len(errored_configs) == 0, f"Errored configs: {errored_configs}"