Merge pull request #39 from wandb/olachinkei-patch-1

Update utils.py
wandb · Oct 29, 2023 · df24458 · df24458
2 parents c00e0a9 + 3928abe
commit df24458
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils.py b/src/utils.py
@@ -222,7 +222,7 @@ def eval_JCoLA(dataset,llm_chain):
     for i in tqdm(range(len(dataset['validation']))):
         sentence = dataset['validation'][i]['sentence']
         y_true = dataset['validation'][i]['label']
-        label = str(y_true)  + ["acceptable: ", "unacceptable: "][y_true]
+        label = str(y_true)  + [" acceptable", " unacceptable"][y_true]
         y_pred = overall_chain({'sentence':sentence, 'label':label},callbacks=[WandbTracer()])['output']
         y_pred = y_pred.strip().lower()