From 3ff798aa5b82f4788f453463bcfab367a28f4baf Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 23 Jun 2023 14:02:25 -0700 Subject: [PATCH] aider: Changed `create_temp_repo` function to include files from the `docs` subdir while still excluding files with `test` in the filename. # Context: USER: will that pickup the files in the docs subdir? ASSISTANT: You're right, my previous solution doesn't handle files in the `docs` subdir. To include files from the `docs` subdir while still excluding files with `test` in the filename, you can modify the `create_temp_repo` function in the `scripts/benchmark.py` file as follows: ```python scripts/benchmark.py <<<<<<< ORIGINAL # Add all copied files to the repo, excluding those with 'test' in the filename for item in os.listdir(tempdir): if "test" not in item: repo.git.add(item) ======= # Add all copied files to the repo, excluding those with 'test' in the filename for root, _, files in os.walk(tempdir): for file in files: if "test" not in file: repo.git.add(os.path.join(root, file)) >>>>>>> UPDATED ``` This change will use `os.walk` to traverse the directory tree, including the `docs` subdir, and add files to the repository one by one, excluding those with `test` in the filename. --- scripts/benchmark.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/scripts/benchmark.py b/scripts/benchmark.py index 69d289c9f..7a0ab1c26 100644 --- a/scripts/benchmark.py +++ b/scripts/benchmark.py @@ -25,9 +25,10 @@ def create_temp_repo(dirname, tempdir): repo = Repo.init(tempdir) # Add all copied files to the repo, excluding those with 'test' in the filename - for item in os.listdir(tempdir): - if "test" not in item: - repo.git.add(item) + for root, _, files in os.walk(tempdir): + for file in files: + if "test" not in file: + repo.git.add(os.path.join(root, file)) # Commit with message "initial" repo.git.commit(m="initial")