Skip to content

Commit cf05bc6

Browse files
recursixgasse
authored andcommitted
Add method to override max_steps in Study class
1 parent c5dfb17 commit cf05bc6

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/agentlab/experiments/study.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,10 @@ def get_report(self, ignore_cache=False, ignore_stale=False):
254254
return inspect_results.get_study_summary(
255255
self.dir, ignore_cache=ignore_cache, ignore_stale=ignore_stale
256256
)
257+
258+
def override_max_steps(self, max_steps):
259+
for exp_args in self.exp_args_list:
260+
exp_args.env_args.max_steps = max_steps
257261

258262
@staticmethod
259263
def load(dir: Path) -> "Study":

0 commit comments

Comments
 (0)