update regex (#940)

argilla-io · Sep 2, 2024 · a2a8e86 · a2a8e86
1 parent d5f2ae3
commit a2a8e86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/distilabel/steps/tasks/sentence_transformers.py b/src/distilabel/steps/tasks/sentence_transformers.py
@@ -33,7 +33,7 @@
 GenerationAction = Literal["paraphrase", "semantically-similar", "query", "answer"]
 
 POSITIVE_NEGATIVE_PAIR_REGEX = re.compile(
-    r"## Positive\s+(.*?)(?:\s+## Negative\s+(.*?))?\s*$",
+    r"\s*## Positive\s+(.*?)(?:\s+## Negative\s+(.*?))?\s*$",
     re.DOTALL,
 )