From 347552760873f0081cdda80bf1acb9c96f1b216b Mon Sep 17 00:00:00 2001 From: Tucker Date: Wed, 13 Jan 2021 22:43:43 -0500 Subject: [PATCH] Add mode to get_best_logdir --- examples/rllib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)