Skip to content

Commit b3f4d7f

Browse files
committed
change the prompt
1 parent a8ebfcc commit b3f4d7f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

code_bert/cli/run_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def _run_model(file_path, file_parser, predictor):
2323
for func_name, func_body, docstr in file_parser.parse_file_and_get_data(file_path):
2424
match, _ = predictor.predict(func_body, docstr)
2525
match_yes = "Yes" if bool(match) == True else "No"
26-
print(f'Function "{func_name}" with Dcostring """{docstr}"""\nDo they match?\n{match_yes}')
26+
print(f'>>> Function "{func_name}" with Dcostring """{docstr}"""\n>>> Do they match?\n{match_yes}')
2727
print("******************************************************************")
2828

2929

0 commit comments

Comments
 (0)