Fix pandas collumn error

huawei-noah · Jan 19, 2021 · 8d564be · 8d564be
1 parent 36fb206
commit 8d564be
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/benchmark/run.py b/benchmark/run.py
@@ -138,7 +138,7 @@ def main(
     )
     analysis = tune.run(**config["run"])
 
-    print(analysis.dataframe(mode="max").head())
+    print(analysis.dataframe().head())
 
 
 def parse_args():

diff --git a/examples/rllib.py b/examples/rllib.py
@@ -137,7 +137,7 @@ def main(
         scheduler=pbt,
     )
 
-    print(analysis.dataframe(mode="max").head())
+    print(analysis.dataframe().head())
 
     best_logdir = Path(analysis.get_best_logdir("episode_reward_max"))
     model_path = best_logdir / "model"

diff --git a/smarts/env/tests/test_learning.py b/smarts/env/tests/test_learning.py
@@ -78,7 +78,7 @@ def test_learning_regression_rllib():
         config=tune_confg,
     )
 
-    df = analysis.dataframe(mode=max)
+    df = analysis.dataframe()
 
     # Lower-bound 95% confidence interval of mean reward after one hour, generated by manual analysis.
     # If you need to update this, run tools/regression_rllib.py.