Fix CLI parsing

NVIDIA · Oct 2, 2024 · a5ef4c1 · a5ef4c1
1 parent 846b744
commit a5ef4c1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/scripts/protein/esm2/esm2_pretrain.py b/scripts/protein/esm2/esm2_pretrain.py
@@ -521,7 +521,6 @@ def main(
         wandb_group=args.wandb_group,
         wandb_id=args.wandb_id,
         wandb_anonymous=args.wandb_anonymous,
-        wandb_version=args.wandb_version,
         wandb_log_model=args.wandb_log_model,
         wandb_offline=args.wandb_offline,
         num_steps=args.num_steps,

diff --git a/scripts/singlecell/geneformer/train.py b/scripts/singlecell/geneformer/train.py
@@ -344,9 +344,7 @@ def main(
 parser.add_argument(
     "--wandb-id", type=str, default=None, help="Sets the version, mainly used to resume a previous run"
 )
-parser.add_argument(
-    "--wandb-anonymous", type=str, action="store_true", help="Enable or explicitly disable anonymous logging"
-)
+parser.add_argument("--wandb-anonymous", action="store_true", help="Enable or explicitly disable anonymous logging")
 parser.add_argument(
     "--wandb-log-model", action="store_true", help="Save checkpoints in wandb dir to upload on W&B servers"
 )
@@ -520,7 +518,6 @@ def config_class_type(desc: str) -> Type[BioBertGenericConfig]:
         wandb_group=args.wandb_group,
         wandb_id=args.wandb_id,
         wandb_anonymous=args.wandb_anonymous,
-        wandb_version=args.wandb_version,
         wandb_log_model=args.wandb_log_model,
         wandb_offline=args.wandb_offline,
         num_steps=args.num_steps,