change: Move conversion into its own function

src/braket/simulator_v2/simulator.py

@@ -18,6 +18,23 @@
 from .julia_import import jl, jlBraketSimulator
 
 
+def _analytic_result_value_to_ndarray(
+    task_result: GateModelTaskResult,
+) -> GateModelTaskResult:
+    """Convert any StateVector or DensityMatrix result values from raw Python lists to the expected
+    np.ndarray. This must be done because the wrapper Julia simulator results Python lists to comply
+    with the pydantic specification for ResultTypeValues.
+    """
+    for result_ind, result_type in enumerate(task_result.resultTypes):
+        if isinstance(result_type.type, StateVector) or isinstance(
+            result_type.type, DensityMatrix
+        ):
+            task_result.resultTypes[result_ind].value = np.asarray(
+                task_result.resultTypes[result_ind].value
+            )
+    return task_result
+
+
 class StateVectorSimulatorV2(BaseLocalSimulator):
     """A state vector simulator meant to run directly on the user's machine using a Julia backend.
 
@@ -92,15 +109,7 @@ def run_jaqcd(
         r = jl.simulate(self._device, [circuit_ir], qubit_count, shots)
         r.additionalMetadata.action = circuit_ir
         if not shots:
-            # need to convert `list` value for `statevector`
-            # and `densitymatrix` result types to `np.ndarray`
-            for result_ind, result_type in enumerate(r.resultTypes):
-                if isinstance(result_type.type, StateVector) or isinstance(
-                    result_type.type, DensityMatrix
-                ):
-                    r.resultTypes[result_ind].value = np.asarray(
-                        r.resultTypes[result_ind].value
-                    )
+            r = _analytic_result_value_to_ndarray(r)
         return r
 
     def run_openqasm(
@@ -166,19 +175,11 @@ def run_openqasm(
             self._device, [circuit], qubit_count, shots, measured_qubits=measured_qubits
         )
         r.additionalMetadata.action = openqasm_ir
-        # attach the result types
         if shots:
+            # attach the result types
             r.resultTypes = results
         else:
-            # need to convert `list` value for `statevector`
-            # and `densitymatrix` result types to `np.ndarray`
-            for result_ind, result_type in enumerate(r.resultTypes):
-                if isinstance(result_type.type, StateVector) or isinstance(
-                    result_type.type, DensityMatrix
-                ):
-                    r.resultTypes[result_ind].value = np.asarray(
-                        r.resultTypes[result_ind].value
-                    )
+            r = _analytic_result_value_to_ndarray(r)
         return r
 
     @property
@@ -497,12 +498,7 @@ def run_jaqcd(
         r = jl.simulate(self._device, [circuit_ir], qubit_count, shots)
         r.additionalMetadata.action = circuit_ir
         if not shots:
-            # need to convert `list` value for `densitymatrix` result type to `np.ndarray`
-            for result_ind, result_type in enumerate(r.resultTypes):
-                if isinstance(result_type.type, DensityMatrix):
-                    r.resultTypes[result_ind].value = np.asarray(
-                        r.resultTypes[result_ind].value
-                    )
+            r = _analytic_result_value_to_ndarray(r)
         return r
 
     def run_openqasm(
@@ -571,12 +567,7 @@ def run_openqasm(
         if shots:
             r.resultTypes = results
         else:
-            # need to convert `list` value for `densitymatrix` result type to `np.ndarray`
-            for result_ind, result_type in enumerate(r.resultTypes):
-                if isinstance(result_type.type, DensityMatrix):
-                    r.resultTypes[result_ind].value = np.asarray(
-                        r.resultTypes[result_ind].value
-                    )
+            r = _analytic_result_value_to_ndarray(r)
         return r
 
     @property