f1afb5df71893ff2770c7cc7ca83d2ac68f977d7,tests/test_models_misc.py,,test_positive_wavefunction_psi,#,149

Before Change

    nn_state = PositiveWaveFunction(10, gpu=False)

    vis_state = torch.ones(10).to(dtype=torch.double)
    actual_psi = nn_state.psi(vis_state)[].to(vis_state)
    expected_psi = torch.zeros(1).to(vis_state)

    msg = "PositiveWaveFunction is giving a non-zero imaginary part!"

After Change

    nn_state = PositiveWaveFunction(10, gpu=False)

    vis_state = torch.ones(10).to(dtype=torch.double)
    actual_psi_im = cplx.imag(nn_state.psi(vis_state)).to(vis_state)
    expected_psi_im = torch.zeros(1).squeeze().to(vis_state)

    msg = "PositiveWaveFunction is giving a non-zero imaginary part!"

Instances