NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-23 16:54:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

add error logging to collect_trajectories so they don't fail silently

Browse source
This commit is contained in:
Dakota 2025-08-15 16:34:21 -05:00
parent 628bd3d2ad
commit 11f1303da0
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
atroposlib/envs/base.py
View file

@ -889,7 +889,8 @@ class BaseEnv(ABC):
# do a rollout with item
try:
to_postprocess, to_backlog = await self.collect_trajectories(item)
except Exception:
except Exception as e:
logging.error(f"Error in collect_trajectories: {e}")
to_postprocess = None
to_backlog = []
# add the items to the queue
@ -911,7 +912,10 @@ class BaseEnv(ABC):
self.task_successful.append(1)
self.succeeded_task_duration.append(duration)
logger.debug(f"handle_env: Collected {len(to_postprocess)} trajectories")
await self.handle_send_to_api(to_postprocess, item)
try:
await self.handle_send_to_api(to_postprocess, item)
except Exception as e:
logger.error(f"Error in handle_send_to_api: {e}")
else:
self.task_successful.append(0)
self.failed_task_duration.append(duration)