def METHOD_NAME(self, model, extra_files=None):
    # Just check that we're able to run successfully.
    buf = io.BytesIO()
    torch.jit.save(model, buf, _extra_files=extra_files)
    info = torch.utils.model_dump.get_model_info(buf)
    assert info is not None