We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d24dfbe commit acc6f97Copy full SHA for acc6f97
tests/models/test_models_abc.py
@@ -174,5 +174,5 @@ def test_get_pretrained_model_not_str() -> None:
174
175
def test_get_pretrained_model_not_in_info() -> None:
176
"""Test ValueError is raised if input is not in info."""
177
- with pytest.raises(ValueError, match=r"Pretrained model `alexnet` does not exist."):
178
- _ = get_pretrained_model("alexnet")
+ with pytest.raises(ValueError, match=r"Pretrained model `random` does not exist."):
+ _ = get_pretrained_model("random")
0 commit comments