Skip to content

Commit 6edfb1a

Browse files
committed
fix: overwriting default user prompt with ac.USER_PROMPT
1 parent 59634b8 commit 6edfb1a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

run_ac.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,8 @@ def parse_data_to_record_dict(record_chunk):
110110
# the script `labeling_functions` does not exist. It will be inserted at runtime
111111
import attribute_calculators
112112

113+
DEFAULT_USER_PROMPT = attribute_calculators.USER_PROMPT
114+
113115
vocab = spacy.blank(iso2_code).vocab
114116

115117
with open("docbin_full.json", "r") as infile:
@@ -127,7 +129,7 @@ def parse_data_to_record_dict(record_chunk):
127129
__print_progress(0.0)
128130
for record_dict in record_dict_list:
129131
attribute_calculators.USER_PROMPT = prepare_and_render_mustache(
130-
attribute_calculators.USER_PROMPT, record_dict
132+
DEFAULT_USER_PROMPT, record_dict
131133
)
132134

133135
idx += 1

0 commit comments

Comments
 (0)