bug

boettiger-lab · Jul 9, 2024 · f725146 · f725146
1 parent c40a3dc
commit f725146
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/train.py b/scripts/train.py
@@ -3,7 +3,7 @@
 parser = argparse.ArgumentParser()
 parser.add_argument("-f", "--file", help="Path config file", type=str)
 parser.add_argument("-pb", "--progress_bar", help="Use  progress bar for training", type=bool, default=True)
-parser.add_argument("-id", "--identifier", help="ID string for saving the agent", type=bool, default="0")
+parser.add_argument("-id", "--identifier", help="ID string for saving the agent", type=str, default="0")
 args = parser.parse_args()
 
 import rl4caribou

diff --git a/src/rl4caribou/utils/sb3.py b/src/rl4caribou/utils/sb3.py
@@ -37,7 +37,7 @@ def algorithm(algo):
     }
     return algos[algo]
 
-def sb3_train(config_file, *, progress_bar, identifier, **kwargs):
+def sb3_train(config_file, progress_bar=True, identifier="0", **kwargs):
     with open(config_file, "r") as stream:
         options = yaml.safe_load(stream)
         options = {**options, **kwargs}