Skip to content

Commit acc6f97

Browse files
committed
✅ Fix test for alexnet raise error check, as timm supports alexnet no error is generated.
1 parent d24dfbe commit acc6f97

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/models/test_models_abc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -174,5 +174,5 @@ def test_get_pretrained_model_not_str() -> None:
174174

175175
def test_get_pretrained_model_not_in_info() -> None:
176176
"""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")
177+
with pytest.raises(ValueError, match=r"Pretrained model `random` does not exist."):
178+
_ = get_pretrained_model("random")

0 commit comments

Comments
 (0)