fix syntax

JetBrains-Research · Sep 24, 2024 · f417a87 · f417a87
1 parent b84f182
commit f417a87
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/verified_cogen/runners/chain_of_thought/step.py b/verified_cogen/runners/chain_of_thought/step.py
@@ -18,13 +18,15 @@ def __init__(self, data: str):
         appending_to = current_question
         for i, line in enumerate(data_lines):
             if line.startswith("===== QUESTION_EXAMPLE ====="):
-                self.substeps.append(Substep("\n".join(current_question), "\n".join(current_answer)))
+                self.substeps.append(
+                    Substep("\n".join(current_question), "\n".join(current_answer))
+                )
                 current_question, current_answer = [], []
                 appending_to = current_question
-            elif line.startswith("===== ANSWER =====")
+            elif line.startswith("===== ANSWER ====="):
                 appending_to = current_answer
             elif line.startswith("===== QUESTION ====="):
-                self.question = "\n".join(data_lines[i + 1:])
+                self.question = "\n".join(data_lines[i + 1 :])
                 break
             else:
                 appending_to.append(line)