From 436d0dd774d41f6aa08e4076c991fa8ead880d32 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Sat, 24 Jun 2023 20:44:37 -0700
Subject: [PATCH] restore the original file, in case we interrupted a prev run

---
 scripts/benchmark.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/scripts/benchmark.py b/scripts/benchmark.py
index 9598d9862..7c8325c09 100644
--- a/scripts/benchmark.py
+++ b/scripts/benchmark.py
@@ -189,6 +189,11 @@ def run_test(testdir, model_name, edit_format, retries, no_test, verbose):
         if "test" not in fname.name and fname.is_file() and fname.name[0] != ".":
             fnames.append(fname)
 
+            # restore the original file, in case we interrupted a prev run
+            # after it had saved changes
+            original_fname = ORIGINAL_DNAME / testdir.name / fname.name
+            shutil.copy(original_fname, fname)
+
     file_list = " ".join(fname.name for fname in fnames)
     instructions = (testdir / ".docs/instructions.md").read_text()
     instructions += (