From 1288901dc518af27a435e4d42a459beb222c0847 Mon Sep 17 00:00:00 2001 From: Marquess Valdez Date: Fri, 18 Oct 2024 10:39:08 -0700 Subject: [PATCH] fix memory_map construction --- test/test_e2e_no_qcs.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/test_e2e_no_qcs.py b/test/test_e2e_no_qcs.py index 94d6adc..a423308 100644 --- a/test/test_e2e_no_qcs.py +++ b/test/test_e2e_no_qcs.py @@ -3,6 +3,7 @@ from typing import Dict, List import numpy as np +from pyquil.api import MemoryMap from pyquil.gates import MEASURE, RX from pyquil.quil import Program from pyquil.quilatom import MemoryReference @@ -71,7 +72,7 @@ def test_run_batch(qc: AzureQuantumComputer) -> None: """Test the ``run_batch`` interface which should be much faster than normal parametrization""" compiled = qc.compile(PARAMETRIZED) - memory_map: Dict[str, List[List[float]]] = {"theta": [[0], [np.pi], [2 * np.pi]]} + memory_map: MemoryMap = {"theta": [0, np.pi, 2 * np.pi]} results = qc.run_batch(compiled, memory_map) results_0 = results[0].get_register_map().get("ro")