diff --git a/spacy/gold/loggers.py b/spacy/gold/loggers.py index 7c5a3d317..9feec558f 100644 --- a/spacy/gold/loggers.py +++ b/spacy/gold/loggers.py @@ -67,7 +67,7 @@ def console_logger(): @registry.loggers("spacy.WandbLogger.v1") -def wandb_logger(project_name: str, disable_fields: list = []): +def wandb_logger(project_name: str, remove_config_values: list = []): import wandb console = console_logger() @@ -77,7 +77,7 @@ def wandb_logger(project_name: str, disable_fields: list = []): ) -> Tuple[Callable[[Dict[str, Any]], None], Callable]: config = nlp.config.interpolate() config_dot = util.dict_to_dot(config) - for field in disable_fields: + for field in remove_config_values: del config_dot[field] config = util.dot_to_dict(config_dot) wandb.init(project=project_name, config=config)