perf: avoid variable overrides by end users

code-kern-ai · Jan 17, 2025 · 26a8b35 · 26a8b35
1 parent 23ecd8a
commit 26a8b35
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/run_ac.py b/run_ac.py
@@ -110,7 +110,7 @@ def parse_data_to_record_dict(record_chunk):
     # the script `labeling_functions` does not exist. It will be inserted at runtime
     import attribute_calculators
 
-    DEFAULT_USER_PROMPT = attribute_calculators.USER_PROMPT
+    DEFAULT_USER_PROMPT_A2VYBG = attribute_calculators.USER_PROMPT_A2VYBG
 
     vocab = spacy.blank(iso2_code).vocab
 
@@ -128,8 +128,8 @@ def parse_data_to_record_dict(record_chunk):
     amount = len(record_dict_list)
     __print_progress(0.0)
     for record_dict in record_dict_list:
-        attribute_calculators.USER_PROMPT = prepare_and_render_mustache(
-            DEFAULT_USER_PROMPT, record_dict
+        attribute_calculators.USER_PROMPT_A2VYBG = prepare_and_render_mustache(
+            DEFAULT_USER_PROMPT_A2VYBG, record_dict
         )
 
         idx += 1