From 17bc4c8b361a4104b73d5599fb3b1dbf96d7bd17 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Mon, 14 Oct 2024 18:03:35 +0700 Subject: [PATCH] fix: update test based on new defaults --- tests/test_validation_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_validation_dataset.py b/tests/test_validation_dataset.py index c9bfebfc2..499f7dbbe 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 == ChatTemplate.tokenizer_default + assert checked_cfg.chat_template == None assert ( checked_cfg.datasets[0].chat_template == ChatTemplate.tokenizer_default )