def METHOD_NAME(self, checkpoint: str) -> None:
    match = "num classes .* != num classes in pretrained model"
    with pytest.warns(UserWarning, match=match):
        SimCLRTask(model="resnet18", weights=checkpoint)