def state_prep_circuit(
    alice_data_qubits: tuple[cirq.GridQubit, cirq.GridQubit],
    bob_data_qubits: tuple[cirq.GridQubit, cirq.GridQubit],
) -> cirq.Circuit:
    """Construct a circuit to produce the initial Bell pairs.

    Args:
        alice_data_qubits: Alice's data qubits.
        bob_data_qubits: Bob's data qubits.

    Returns:
        A circuit preparing the Bell states.
    """
    a1, a2 = alice_data_qubits
    b1, b2 = bob_data_qubits
    return bell_pair_prep_circuit(a1, b1).zip(bell_pair_prep_circuit(a2, b2))
Construct a circuit to produce the initial Bell pairs.

    Args:
        alice_data_qubits: Alice's data qubits.
        bob_data_qubits: Bob's data qubits.

    Returns:
        A circuit preparing the Bell states.
    
examples/magic_square.py