[HOTFIX] Fix usage example notebooks for eval callback

ugr-sail · Jun 27, 2024 · 00093d0 · 00093d0
1 parent 9c37c2d
commit 00093d0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/drl.ipynb b/examples/drl.ipynb
@@ -380,7 +380,8 @@
                 "\n",
                 "# Set up Evaluation and saving best model\n",
                 "eval_callback = LoggerEvalCallback(\n",
-                "    eval_env,\n",
+                "    eval_env=eval_env,\n",
+                "    train_env=env,\n",
                 "    best_model_save_path=eval_env.get_wrapper_attr('workspace_path') +\n",
                 "    '/best_model/',\n",
                 "    log_path=eval_env.get_wrapper_attr('workspace_path') +\n",
@@ -80656,7 +80657,7 @@
             "name": "python",
             "nbconvert_exporter": "python",
             "pygments_lexer": "ipython3",
-            "version": "3.10.12"
+            "version": "3.11.2"
         },
         "vscode": {
             "interpreter": {