mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-24 17:04:55 +00:00
fix: replace debug print statements with logger in dataset_env and infinimath_env
This commit is contained in:
parent
7da681ec46
commit
1303cb59e8
2 changed files with 2 additions and 2 deletions
|
|
@ -539,7 +539,7 @@ class InfiniteMathEnv(BaseEnv):
|
|||
if hasattr(completion, "text")
|
||||
else completion.message.content
|
||||
)
|
||||
print("model_answer", model_answer)
|
||||
logger.debug(f"model_answer: {model_answer}")
|
||||
|
||||
full_messages = [
|
||||
{"role": "system", "content": self.system_prompt},
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue