From 60f7f71622471cf4a562ef7d2c6a04af14887b4e Mon Sep 17 00:00:00 2001
From: Vincent Michaud-Rioux <vincent.michaud-rioux@xanadu.ai>
Date: Thu, 26 Oct 2023 15:09:54 +0000
Subject: [PATCH] Fix yml format

---
 .github/workflows/tests_linux_x86_mpi_gpu.yml | 8 ++++----
 mpitests/test_adjoint_jacobian.py             | 3 ++-
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/tests_linux_x86_mpi_gpu.yml b/.github/workflows/tests_linux_x86_mpi_gpu.yml
index 744f34a964..4d277a73ac 100644
--- a/.github/workflows/tests_linux_x86_mpi_gpu.yml
+++ b/.github/workflows/tests_linux_x86_mpi_gpu.yml
@@ -244,10 +244,10 @@ jobs:
         run: |
           source /etc/profile.d/modules.sh && module use /opt/modules/ && module load ${{ matrix.mpilib }}
           cat > runscript<<EOF
-python -m pytest ./mpitests \
---cov=pennylane_lightning --no-flaky-report -p no:warnings --tb=native --cov-report=term-missing \
---cov-report=xml:./coverage-${{ github.job }}-lightning_gpu_${{ matrix.mpilib }}-\$OMPI_COMM_WORLD_RANK.xml
-EOF
+          python -m pytest ./mpitests \
+          --cov=pennylane_lightning --no-flaky-report -p no:warnings --tb=native --cov-report=term-missing \
+          --cov-report=xml:./coverage-${{ github.job }}-lightning_gpu_${{ matrix.mpilib }}-\$OMPI_COMM_WORLD_RANK.xml
+          EOF
           cat runscript
           PL_DEVICE=lightning.gpu /opt/mpi/${{ matrix.mpilib }}/bin/mpirun -np 2 runscript
           # mv coverage_lgpu_mpi.xml coverage-${{ github.job }}-lightning_gpu_${{ matrix.mpilib }}-main.xml
diff --git a/mpitests/test_adjoint_jacobian.py b/mpitests/test_adjoint_jacobian.py
index 166ada4d3e..a169426462 100644
--- a/mpitests/test_adjoint_jacobian.py
+++ b/mpitests/test_adjoint_jacobian.py
@@ -1212,12 +1212,13 @@ def circuit_compare(params, excitations):
 
     # np.random.seed(1337)
     # params = 1.0e-3 * np.random.rand(len(doubles))
-    # comm = MPI.COMM_WORLD
+    comm = MPI.COMM_WORLD
     # params = comm.bcast(params, root=0)
     params = qml.numpy.array([0.0] * len(doubles), requires_grad=True)
 
     jacs = jac_func(params, excitations=doubles)
     jacs_comp = jac_func_comp(params, excitations=doubles)
+    comm.Barrier()
 
     assert np.allclose(jacs, jacs_comp)