From 41e13ce490cc62a3cccd72eb145c20b63d0735d0 Mon Sep 17 00:00:00 2001
From: Alexei-V-Ivanov-AMD
 <156011006+Alexei-V-Ivanov-AMD@users.noreply.github.com>
Date: Mon, 4 Mar 2024 14:58:29 -0600
Subject: [PATCH] Fixing an apparent typo in process_humaneval.py

---
 WizardCoder/src/process_humaneval.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/WizardCoder/src/process_humaneval.py b/WizardCoder/src/process_humaneval.py
index 1023a09..5a687ed 100644
--- a/WizardCoder/src/process_humaneval.py
+++ b/WizardCoder/src/process_humaneval.py
@@ -51,7 +51,7 @@
                     print("================\n")
                 # print(completion)
             if "__name__ == \"__main__\"" in completion:
-                next_line = completion.index('if __name__ == "__main__":')
+                next_line = completion.index('__name__ == "__main__"')
                 completion = completion[:next_line].strip()
                 # print(completion)
             
@@ -66,4 +66,4 @@
     
 print("save to {}".format(args.out_path))
 write_jsonl(args.out_path, output)
-print(a)
\ No newline at end of file
+print(a)