Skip to content

Commit 546431e

Browse files
committed
fix stats calculation for max_iterations
1 parent 4c4b980 commit 546431e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

agents/attackers/llm_qa/llm_agent_qa.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,8 @@
234234
reach_max_steps += 1
235235
type_of_end = "max_iterations"
236236
total_reward = -100
237-
steps = observation.info["max_steps"]
237+
#steps = observation.info["max_steps"] #this fails
238+
steps = num_iterations
238239
else:
239240
reach_max_steps += 1
240241
type_of_end = "max_steps"

0 commit comments

Comments
 (0)