NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2025-08-07 13:37:48 -04:00
parent 62b72589c6
commit f76f9d1596
1 changed files with 2 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

11
environments/eval_environments/refusalbench_environment.py
View file

@ -912,13 +912,7 @@ Your answer:"""
overarching_category = eval_item.get("overarching_category", "Unknown") overarching_category = eval_item.get("overarching_category", "Unknown")
danger = eval_item.get("danger", "unknown") danger = eval_item.get("danger", "unknown")
print("Here is the prompt text ", prompt_text)
print("Here is the overarching category ", overarching_category)
print("Here is the danger ", danger)
if not prompt_text: if not prompt_text:
return {"refusal_rate": None, "sample": None} return {"refusal_rate": None, "sample": None}
@ -943,11 +937,10 @@ Your answer:"""
completion_params, completion_params,
f"EVAL attempt {attempt + 1}/{max_retries}", f"EVAL attempt {attempt + 1}/{max_retries}",
) )
print("Tried")
completion = await self.server.chat_completion( completion = await self.server.chat_completion(
messages=messages, **completion_params messages=messages, **completion_params
) )
print("passed")
# Log full debug response # Log full debug response
self._log_full_debug_response( self._log_full_debug_response(