Skip to content

Commit

Permalink
Testing prints
Browse files Browse the repository at this point in the history
  • Loading branch information
jamesbraza committed Sep 13, 2024
1 parent 5361849 commit b56e3c5
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions paperqa/agents/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -371,10 +371,12 @@ async def run_ldp_agent(

while not done:
action, agent_state, value = await agent.get_asv(agent_state, obs)
print(action.value)
if on_agent_action_callback:
await on_agent_action_callback(action, agent_state, value)

obs, reward, done, truncated = await env.step(action.value)
print(obs)
if on_env_step_callback:
await on_env_step_callback(obs, reward, done, truncated)
status = AgentStatus.SUCCESS
Expand Down

0 comments on commit b56e3c5

Please sign in to comment.