test_random_ansatz.py

from src import Individual
import src.myflow as myflow
from src.args import args
from random import seed
import numpy as np
from src.uuid import RUN_UUID
from src import save_circuit


# SEED = 91169 # Very bad
# SEED = 9116  # Quite good
SEED = 42
_SAVE_DIR = "Best_ansatz"


def main():
    myflow.connect(
        run_name=args.run_name,
        nolog=args.nomlflow,
        log_system_metrics=args.log_system_metrics,
    )
    myflow.log_params(vars(args))
    myflow.log_params(
        {
            "UUID": RUN_UUID,
            "SEED": SEED,
        }
    )

    seed(SEED)
    np.random.seed(SEED)

    # generate 10 random Ansatz
    for i in range(10):
        random_ansatz = Individual()
        print(random_ansatz)

        # run VQA with POWELL optimizer
        _fitness = random_ansatz.fitness
        # myflow.log_metric("best_fitness", _fitness, step=i)
        myflow.log_metric(f"fitness_{i}", _fitness)
        print(f"Fitness: {_fitness}")

        save_circuit(
            random_ansatz,
            _SAVE_DIR,
            f"best_ansatz_{i}_{RUN_UUID}",
        )
        myflow.log_artifact(
            f"{_SAVE_DIR}/best_ansatz_{i}_{RUN_UUID}.png",
            artifact_name="results",
        )


if __name__ == "__main__":
    main()