fix test_hf_import test_hf_import_missing_author (kubeflow#287)

likely due to HF maintenance of model ids located at the root-level
dhirajsb · Feb 2, 2024 · 3c6b279 · 3c6b279
1 parent 13a24ab
commit 3c6b279
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/clients/python/tests/test_client.py b/clients/python/tests/test_client.py
@@ -70,7 +70,7 @@ def test_get(mr_client: ModelRegistry):
 
 def test_hf_import(mr_client: ModelRegistry):
     pytest.importorskip("huggingface_hub")
-    name = "gpt2"
+    name = "openai-community/gpt2"
     version = "1.2.3"
 
     assert mr_client.register_hf_model(
@@ -87,13 +87,13 @@ def test_hf_import(mr_client: ModelRegistry):
 
 def test_hf_import_missing_author(mr_client: ModelRegistry):
     pytest.importorskip("huggingface_hub")
-    name = "gpt2"
+    name = "bert-base-uncased"
     version = "1.2.3"
 
     with pytest.warns(match=r".*author is unknown.*"):
         assert mr_client.register_hf_model(
             name,
-            "onnx/decoder_model.onnx",
+            "model.onnx",
             version=version,
             model_format_name="test format",
             model_format_version="test version",