diff --git a/tests/test_validation_dataset.py b/tests/test_validation_dataset.py index 499f7dbbe..389424217 100644 --- a/tests/test_validation_dataset.py +++ b/tests/test_validation_dataset.py @@ -100,7 +100,7 @@ class TestValidationCheckDatasetConfig(BaseValidation): def _check_config(): assert checked_cfg.datasets[0].path == cfg.datasets[0].path assert checked_cfg.datasets[0].type == cfg.datasets[0].type - assert checked_cfg.chat_template == None + assert checked_cfg.chat_template is None assert ( checked_cfg.datasets[0].chat_template == ChatTemplate.tokenizer_default )