From 4626f740d73da20f0423f47316f33f973f9558b8 Mon Sep 17 00:00:00 2001 From: rangoliu Date: Wed, 18 Oct 2023 16:55:10 +0800 Subject: [PATCH] [Fix] Fix test tokenizer (#2050) * replace test text * update mmcv version --- mmagic/__init__.py | 2 +- tests/test_models/test_archs/test_tokenizer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mmagic/__init__.py b/mmagic/__init__.py index 9112799de2..cb958d4dce 100644 --- a/mmagic/__init__.py +++ b/mmagic/__init__.py @@ -21,7 +21,7 @@ def digit_version(version_str): MMCV_MIN = '2.0.0' -MMCV_MAX = '2.1.0' +MMCV_MAX = '2.2.0' mmcv_min_version = digit_version(MMCV_MIN) mmcv_max_version = digit_version(MMCV_MAX) mmcv_version = digit_version(mmcv.__version__) diff --git a/tests/test_models/test_archs/test_tokenizer.py b/tests/test_models/test_archs/test_tokenizer.py index f9d5e2367e..30139ec98c 100644 --- a/tests/test_models/test_archs/test_tokenizer.py +++ b/tests/test_models/test_archs/test_tokenizer.py @@ -25,7 +25,7 @@ def setUp(self): def test_encode_and_decode_and_call(self): # test single token - text = 'hello world, kwaheri!' + text = 'Nice bro, kwaheri!' input_ids = self.tokenizer.encode(text).input_ids self.assertEqual(input_ids[-3:-2], self.tokenizer.encode('kwaheri').input_ids[1:-1])