print steps while testing

d3group · Sep 4, 2024 · 9b6d1e2 · 9b6d1e2
1 parent f297ea4
commit 9b6d1e2
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/ddopnew/experiment_functions.py b/ddopnew/experiment_functions.py
@@ -251,6 +251,7 @@ def run_test_episode(   env: BaseEnvironment, # Any environment inheriting from
             step += 1
             sys.stdout.write(f"\rStep {step}")
             sys.stdout.flush()
+
     if eval_step_info:
         print()
 

diff --git a/ddopnew/meta_experiment_functions.py b/ddopnew/meta_experiment_functions.py
@@ -279,6 +279,7 @@ def prep_and_run_test(
     save_dataset: bool = True,
     save_features: bool = False,
     dataset_dir: str = None,
+    eval_step_info = False,
     tracking = "wandb"):
 
     """
@@ -305,6 +306,7 @@ def prep_and_run_test(
         return_dataset=save_dataset,
         save_features=save_features,
         tracking=tracking,
+        eval_step_info=eval_step_info,
     )
 
     # Save dataset

diff --git a/nbs/30_experiment_functions/10_experiment_functions.ipynb b/nbs/30_experiment_functions/10_experiment_functions.ipynb
@@ -436,6 +436,7 @@
     "            step += 1\n",
     "            sys.stdout.write(f\"\\rStep {step}\")\n",
     "            sys.stdout.flush()\n",
+    "\n",
     "    if eval_step_info:\n",
     "        print()\n",
     "\n",

diff --git a/nbs/30_experiment_functions/20_meta_experiment_functions.ipynb b/nbs/30_experiment_functions/20_meta_experiment_functions.ipynb
@@ -459,6 +459,7 @@
     "    save_dataset: bool = True,\n",
     "    save_features: bool = False,\n",
     "    dataset_dir: str = None,\n",
+    "    eval_step_info = False,\n",
     "    tracking = \"wandb\"):\n",
     "    \n",
     "    \"\"\"\n",
@@ -485,6 +486,7 @@
     "        return_dataset=save_dataset,\n",
     "        save_features=save_features,\n",
     "        tracking=tracking,\n",
+    "        eval_step_info=eval_step_info,\n",
     "    )\n",
     "\n",
     "    # Save dataset\n",