chore: cancel cp to bool (#220)

Signed-off-by: ThibaultFy <[email protected]>
Substra · Jun 20, 2024 · 8d9ea87 · 8d9ea87
1 parent 4c201f8
commit 8d9ea87
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/benchmark/camelyon/common/utils.py b/benchmark/camelyon/common/utils.py
@@ -75,7 +75,12 @@ def parse_params() -> dict:
         default=Path(__file__).resolve().parents[1] / "data",
         help="Path to the data",
     )
-    parser.add_argument("--cancel-cp", type=bool, default=False, help="Remote only: cancel the CP after registration")
+    parser.add_argument(
+        "--cancel-cp",
+        action="store_true",
+        default=False,
+        help="Remote only: cancel the CP after registration",
+    )
     parser.add_argument("--torch-gpu", action="store_true", help="Use PyTorch with GPU/CUDA support")
     parser.add_argument(
         "--skip-pure-torch",