From dee5c448237b3ce4975b580b0890878db9daa342 Mon Sep 17 00:00:00 2001
From: Maximilian Schmidt <maximilian.schmidt@ims.uni-stuttgart.de>
Date: Thu, 4 Apr 2024 10:32:17 +0200
Subject: [PATCH] Improve code readability
---
utils.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/utils.py b/utils.py
index 691b3bd..167451a 100644
--- a/utils.py
+++ b/utils.py
@@ -39,8 +39,8 @@ def initialize_run_directory(model: OpenAI | Llama2):
response, usage = model(None, run_name_prompt, chat=True)
model.usage -= usage
run_name_match = re.search(r"^\w+$", response, re.MULTILINE)
- existing_dirs = os.listdir(RUNS_DIR)
- if run_name_match is None or run_name_match.group(0) in existing_dirs:
+ existing_run_names = os.listdir(RUNS_DIR)
+ if run_name_match is None or run_name_match.group(0) in existing_run_names:
run_name = uuid4().hex
else:
run_name = run_name_match.group(0)
--
GitLab