workaround for argparse boolean variable

dfki-ric-underactuated-lab · Jan 11, 2024 · e130779 · e130779
1 parent b076f57
commit e130779
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/leaderboard/real_hardware/acrobot/create_leaderboard.py b/leaderboard/real_hardware/acrobot/create_leaderboard.py
@@ -28,7 +28,7 @@
     help="Whether to force the recomputation of the leaderboard even without new data.",
     default=False,
     required=False,
-    type=bool,
+    type=int,
 )
 parser.add_argument(
     "--link-base",
@@ -41,7 +41,7 @@
 
 data_dir = parser.parse_args().data_dir
 save_to = parser.parse_args().save_to
-recompute_leaderboard = parser.parse_args().recompute
+recompute_leaderboard = bool(parser.parse_args().recompute)
 link_base = parser.parse_args().link
 
 if not os.path.exists(save_to):

diff --git a/leaderboard/real_hardware/pendubot/create_leaderboard.py b/leaderboard/real_hardware/pendubot/create_leaderboard.py
@@ -28,7 +28,7 @@
     help="Whether to force the recomputation of the leaderboard even without new data.",
     default=False,
     required=False,
-    type=bool,
+    type=int,
 )
 parser.add_argument(
     "--link-base",
@@ -41,7 +41,7 @@
 
 data_dir = parser.parse_args().data_dir
 save_to = parser.parse_args().save_to
-recompute_leaderboard = parser.parse_args().recompute
+recompute_leaderboard = bool(parser.parse_args().recompute)
 link_base = parser.parse_args().link
 
 if not os.path.exists(save_to):