From 1b27ecc4ab502714340b0a524e163d9563b02ab9 Mon Sep 17 00:00:00 2001 From: MingLiangDai <95396491+MingLiangDai@users.noreply.github.com> Date: Thu, 30 Nov 2023 16:20:39 -0500 Subject: [PATCH] test: remove outdated tests with pretrained_class parameter (#4308) remove outdated tests with pretrained_class parameter --- .../frameworks/test_transformers_unit.py | 27 ------------------- 1 file changed, 27 deletions(-) diff --git a/tests/integration/frameworks/test_transformers_unit.py b/tests/integration/frameworks/test_transformers_unit.py index 0849ade2f16..ea351047260 100644 --- a/tests/integration/frameworks/test_transformers_unit.py +++ b/tests/integration/frameworks/test_transformers_unit.py @@ -344,30 +344,3 @@ def test_import_model_has_required_files(): assert os.path.exists(bento_model.path_of("model.yaml")) assert bento_model.info.labels == labels bentoml.models.delete("tiny-bert:3956d303d3cddf0708ff20660c1ea5f6ec30e434") - - -def test_import_model_with_pretrained_class(): - revision = "3956d303d3cddf0708ff20660c1ea5f6ec30e434" - pretrained_class = transformers.BertForMaskedLM - bento_model = bentoml.transformers.import_model( - "tiny-bert", - "hf-internal-testing/tiny-random-BertModel", - sync_with_hub_version=True, - pretrained_model_class=pretrained_class, - revision=revision, - ) - - assert bento_model.info.metadata["_pretrained_class"] == pretrained_class.__name__ - bentoml.models.delete("tiny-bert:3956d303d3cddf0708ff20660c1ea5f6ec30e434") - - pretrained_class = transformers.BertForNextSentencePrediction - bento_model = bentoml.transformers.import_model( - "tiny-bert", - "hf-internal-testing/tiny-random-BertModel", - sync_with_hub_version=True, - pretrained_model_class=pretrained_class, - revision=revision, - ) - - assert bento_model.info.metadata["_pretrained_class"] == pretrained_class.__name__ - bentoml.models.delete("tiny-bert:3956d303d3cddf0708ff20660c1ea5f6ec30e434")