Update tests/models/vitpose/test_modeling_vitpose.py

Co-authored-by: amyeroberts <[email protected]>
NielsRogge · Sep 3, 2024 · 26ee67f · 26ee67f
1 parent 7d82ba6
commit 26ee67f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/vitpose/test_modeling_vitpose.py b/tests/models/vitpose/test_modeling_vitpose.py
@@ -175,7 +175,7 @@ def test_model_common_attributes(self):
     def test_inputs_embeds(self):
         pass
 
-    @unittest.skip(reason="ViTPoseBackbone does not support input and output embeddings")
+    @unittest.skip(reason="ViTPose does not support input and output embeddings")
     def test_model_get_set_embeddings(self):
         pass