From 9a6393d4274007bdcacf5fc633419792037ea120 Mon Sep 17 00:00:00 2001
From: Will Shanks <willshanks@us.ibm.com>
Date: Mon, 14 Oct 2024 16:40:07 -0400
Subject: [PATCH] Fix BackendSamplerV2 level 1 data format

---
 test/base.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/base.py b/test/base.py
index f696ae0e94..2a0ea16388 100644
--- a/test/base.py
+++ b/test/base.py
@@ -559,6 +559,7 @@ def _patched_postprocess_pub(
     elif meas_level == 1:
         raw = np.array(result_memory)
         cplx = raw[..., 0] + 1j * raw[..., 1]
+        cplx = np.reshape(cplx, (*shape, *cplx.shape[1:]))
         meas = {
             item.creg_name: cplx for item in meas_info
         }