From 8912a92449f9a9352710815e1a1ab3851e5f7d19 Mon Sep 17 00:00:00 2001 From: shademe Date: Fri, 23 Dec 2022 14:22:03 +0100 Subject: [PATCH] Explicitly check for `None` when validating progress bar type --- spacy/training/loggers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/training/loggers.py b/spacy/training/loggers.py index 90214d0b5..7de31822e 100644 --- a/spacy/training/loggers.py +++ b/spacy/training/loggers.py @@ -107,8 +107,8 @@ def console_logger_v3( write(msg.row(table_header, widths=table_widths, spacing=spacing)) write(msg.row(["-" * width for width in table_widths], spacing=spacing)) progress = None - expected_progress_types = ("train", "eval", None) - if progress_bar not in expected_progress_types: + expected_progress_types = ("train", "eval") + if progress_bar is not None and progress_bar not in expected_progress_types: raise ValueError( Errors.E1048.format( unexpected=progress_bar, expected=expected_progress_types