Skip to content

Commit

Permalink
remove test_hf_import_missing_author test
Browse files Browse the repository at this point in the history
  • Loading branch information
tarilabs authored and isinyaaa committed Feb 13, 2024
1 parent 44d87e4 commit 3a38263
Showing 1 changed file with 0 additions and 17 deletions.
17 changes: 0 additions & 17 deletions clients/python/tests/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,20 +83,3 @@ def test_hf_import(mr_client: ModelRegistry):
)
assert mr_client.get_model_version(name, version)
assert mr_client.get_model_artifact(name, version)


def test_hf_import_missing_author(mr_client: ModelRegistry):
pytest.importorskip("huggingface_hub")
name = "bert-base-uncased"
version = "1.2.3"

with pytest.warns(match=r".*author is unknown.*"):
assert mr_client.register_hf_model(
name,
"model.onnx",
version=version,
model_format_name="test format",
model_format_version="test version",
)
assert (mv := mr_client.get_model_version(name, version))
assert mv.author == "unknown"

0 comments on commit 3a38263

Please sign in to comment.