def METHOD_NAME():
    ha, sx, ma, sampler, driver = _setup_vmc(sr=True)
    sr = driver.preconditioner
    _sr = lambda vstate, grad: sr(vstate, grad)
    with pytest.warns(FutureWarning):
        driver.preconditioner = _sr
    driver.run(1)