f1afb5df71893ff2770c7cc7ca83d2ac68f977d7,tests/test_observables.py,MockWaveFunction,psi,#MockWaveFunction#Any#,86

Before Change

    def psi(self, v):
        // vector/tensor of shape (len(v),)
        amplitude = self.amplitude(v)

        // complex vector; shape: (2, len(v))
        psi = torch.zeros((2,) + amplitude.shape).to(
            dtype=torch.double, device=self.device
        )
        psi[0] = amplitude

        // squeeze down to complex scalar if there was only one visible state
        return psi.squeeze()

After Change

    def psi(self, v):
        // vector/tensor of shape (len(v),)
        return cplx.make_complex(self.amplitude(v))


@pytest.fixture(scope="module")

Instances