From 33883aa764daf71a2e4b50ee26e324beeba488ba Mon Sep 17 00:00:00 2001 From: svlandeg Date: Fri, 28 Aug 2020 14:06:23 +0200 Subject: [PATCH] rename field --- spacy/gold/loggers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)