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

fix: replace debug print statements with logger in dataset_env and infinimath_env

Browse source
This commit is contained in:
alireza78a 2026-02-07 14:51:33 +00:00
parent 7da681ec46
commit 1303cb59e8
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
environments/dataset_environment/dataset_env.py
View file

@ -241,7 +241,7 @@ class DatasetEnv(BaseEnv):
top_p=self.config.top_p,
)
print(f"Completions: {completions}")
logger.debug(f"Completions: {completions}")
# Process completions
trajectories = []
for completion in completions.choices:

2
environments/infinimath/infinimath_env.py
View file

@ -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},