From 588f12c09c6dd6be87163a35713c267ed253d234 Mon Sep 17 00:00:00 2001 From: Lucain Pouget Date: Fri, 14 Jun 2024 16:16:16 +0200 Subject: [PATCH] fix test after cherry --- tests/test_hub_mixin_pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_hub_mixin_pytorch.py b/tests/test_hub_mixin_pytorch.py index 8cc2948593..489af1a37c 100644 --- a/tests/test_hub_mixin_pytorch.py +++ b/tests/test_hub_mixin_pytorch.py @@ -325,7 +325,7 @@ def test_generate_model_card(self): assert card.data.tags == ["model_hub_mixin", "pytorch_model_hub_mixin", "tag1", "tag2"] # Model card template has been used - assert "This is a dummy model card." in str(card) + assert "This is a dummy model card with kwargs." in str(card) model.save_pretrained(self.cache_dir) card_reloaded = ModelCard.load(self.cache_dir / "README.md")