from backend_suite import backends, large_backends
from circuit_suite import circuits

qc = circuits["QAOA_Vanilla"].circuit_lambda(4)
qc.remove_final_measurements()
qc.draw(output="mpl")

from qiskit.converters import circuit_to_dag
from qiskit.visualization import dag_drawer

new_qc = backends["Hatlab-Large-Riswap-Shuffle"].pass_manager.run(qc)
dag = circuit_to_dag(new_qc)
dag_drawer(dag)