Update src/backend/run_toxicity_eval.py

src/backend/run_toxicity_eval.py

@@ -169,7 +169,7 @@ def main(endpoint_url, eval_request):
     prompts = [row['text'] for row in ds['train']['prompt']]
     # All the generated responses from the endpoint
     with Pool() as pool:
-        generated_responses = pool.map(get_generation(endpoint_url, x) for x in prompts[:DATASET_CUTOFF])
+        generated_responses = pool.map([get_generation(endpoint_url, x) for x in prompts[:DATASET_CUTOFF]])
     att_scores_out = score_generations(prompts, generated_responses)
     logger.debug("Scores are:")
     logger.debug(att_scores_out)