diff --git a/examples/rllib.py b/examples/rllib.py index fe24592304..21aca54b7b 100644 --- a/examples/rllib.py +++ b/examples/rllib.py @@ -139,7 +139,7 @@ def main( print(analysis.dataframe().head()) - best_logdir = Path(analysis.get_best_logdir("episode_reward_max")) + best_logdir = Path(analysis.get_best_logdir("episode_reward_max", mode="max")) model_path = best_logdir / "model" copy_tree(str(model_path), save_model_path, overwrite=True)