fix tiny mistake

ASSERT-KTH · Aug 17, 2023 · 5dbe080 · 5dbe080
1 parent 5f1e7f7
commit 5dbe080
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/elleelleaime/sample/prompting/strategies/zero_shot_cloze.py b/elleelleaime/sample/prompting/strategies/zero_shot_cloze.py
@@ -87,13 +87,13 @@ def cloze_prompt(
 
             buggy_file_path = os.path.join(
                 buggy_path,
-                diff[0][2:]
+                diff[0].target_file[2:]
                 if diff[0].target_file.startswith("b/")
                 else diff[0].target_file,
             )
             fixed_file_path = os.path.join(
                 fixed_path,
-                diff[0][2:]
+                diff[0].source_file[2:]
                 if diff[0].source_file.startswith("a/")
                 else diff[0].source_file,
             )