fa244bb56b826ddec459c8257b10308ab945f814,tests/test_training.py,,test_trainingpositive,#,209

Before Change

        nn_state = PositiveWaveFunction(num_visible=nv, num_hidden=nh, gpu=False)

        space = nn_state.generate_hilbert_space(nv)
        callbacks = [
            MetricEvaluator(
                log_every,
                {"Fidelity": ts.fidelity, "KL": ts.KL},
                target=target_psi,
                space=space,
                verbose=True,
            )
        ]

        initialize_posreal_params(request, nn_state)

        nn_state.fit(

After Change

        fidelities.append(ts.fidelity(nn_state, target_psi, space))
        KLs.append(ts.KL(nn_state, target_psi, space))
        print(f"Fidelity: {fidelities[-1]}; KL: {KLs[-1]}.")

    print("\nStatistics")
    print("----------")
    print(

Instances