Fix check on wait_all for bell corrections

QuTech-Delft · Feb 21, 2024 · 9e305e6 · 9e305e6
1 parent 9043330
commit 9e305e6
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/netqasm/sdk/builder.py b/netqasm/sdk/builder.py
@@ -1526,8 +1526,7 @@ def _build_cmds_epr_recv_keep(
 
         self.subrt_add_pending_commands(wait_cmds)  # type: ignore
 
-        # if wait_all and (params.expect_phi_plus or params.expect_psi_plus):
-        if params.expect_phi_plus or params.expect_psi_plus:
+        if wait_all and (params.expect_phi_plus or params.expect_psi_plus):
             self._build_cmds_epr_keep_corrections(
                 qubit_ids_array, ent_results_array, params
             )
@@ -1667,8 +1666,7 @@ def _build_cmds_epr_recv_rsp(
 
         self.subrt_add_pending_commands(wait_cmds)  # type: ignore
 
-        # if wait_all and (params.expect_phi_plus or params.expect_psi_plus):
-        if params.expect_phi_plus or params.expect_psi_plus:
+        if wait_all and (params.expect_phi_plus or params.expect_psi_plus):
             self._build_cmds_epr_keep_corrections(
                 qubit_ids_array, ent_results_array, params
             )