update: search by match (#1096)

argilla-io · Jan 14, 2025 · e866345 · e866345
1 parent 9506930
commit e866345
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/distilabel/steps/tasks/complexity_scorer.py b/src/distilabel/steps/tasks/complexity_scorer.py
@@ -185,7 +185,7 @@ def format_output(
         scores = []
         score_lines = output.split("\n")
         for i, line in enumerate(score_lines):
-            match = _PARSE_SCORE_LINE_REGEX.match(line)
+            match = _PARSE_SCORE_LINE_REGEX.search(line)
             score = float(match.group(1)) if match else None
             scores.append(score)
             if i == len(input["instructions"]) - 1:

diff --git a/src/distilabel/steps/tasks/sentence_transformers.py b/src/distilabel/steps/tasks/sentence_transformers.py
@@ -346,7 +346,7 @@ def format_output(
         if self.use_default_structured_output:
             return self._format_structured_output(output)
 
-        match = POSITIVE_NEGATIVE_PAIR_REGEX.match(output)
+        match = POSITIVE_NEGATIVE_PAIR_REGEX.search(output)
         if match is None:
             formatted_output = {"positive": None}
             if self.triplet: