def cast_result_types()

in src/braket/tasks/gate_model_quantum_task_result.py [0:0]

• 11 lines of code
• 8 McCabe index (conditional complexity)

    def cast_result_types(gate_model_task_result: GateModelTaskResult) -> None:
        """
        Casts the result types to the types expected by the SDK.

        Args:
            gate_model_task_result (GateModelTaskResult): GateModelTaskResult representing the
                results.
        """
        if gate_model_task_result.resultTypes:
            for result_type in gate_model_task_result.resultTypes:
                type = result_type.type.type
                if type == "probability":
                    result_type.value = np.array(result_type.value)
                elif type == "statevector":
                    result_type.value = np.array([complex(*value) for value in result_type.value])
                elif type == "amplitude":
                    for state in result_type.value:
                        result_type.value[state] = complex(*result_type.value[state])