From a5ef4c16f7e6d4926c164f4245691bba070e230d Mon Sep 17 00:00:00 2001 From: John St John Date: Wed, 2 Oct 2024 18:36:52 +0000 Subject: [PATCH] Fix CLI parsing --- scripts/protein/esm2/esm2_pretrain.py | 1 - scripts/singlecell/geneformer/train.py | 5 +---- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/scripts/protein/esm2/esm2_pretrain.py b/scripts/protein/esm2/esm2_pretrain.py index 1a47c6c08b..07e6f658fb 100644 --- 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 index e5236306b6..64a449f7e1 100644 --- 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,