# Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
# http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file is
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.
from enum import Enum
from pydantic.v1 import BaseModel
from braket.ir.jaqcd.shared_models import (
MultiState,
Observable,
OptionalMultiParameter,
OptionalMultiTarget,
OptionalNestedMultiTarget,
)
[docs]
class Expectation(OptionalMultiTarget, Observable):
"""
Expectation of specified targets and observable as requested result.
If no targets are specified, the observable must only operate on 1 qubit and it
will be applied to all qubits in parallel. Otherwise, the number of specified targets
must be equivalent to the number of qubits the observable can be applied to.
Attributes:
type (str): The result type. default = "expectation". (type) is optional.
This should be unique among all result types.
targets (Optional[List[int]]): The target qubits. This is a list of int >= 0.
observable (List[Union[str, List[List[List[float]]]]): A list with at least
one item and items are strings matching the observable regex
or a two dimensional hermitian matrix with complex entries.
Each complex number is represented using a List[float] of size 2, with
element[0] being the real part and element[1] imaginary.
inf, -inf, and NaN are not allowable inputs for the element.
Examples:
>>> Expectation(targets=[1], observable=["x"])
"""
[docs]
class Type(str, Enum):
expectation = "expectation"
type = Type.expectation
[docs]
class AdjointGradient(OptionalNestedMultiTarget, Observable, OptionalMultiParameter):
"""
Adjoint Gradient as requested result.
Attributes:
type (str): The result type. default = "adjoint_gradient". (type) is optional.
This should be unique among all result types.
targets (Optional[List[List[int]]]): The target qubits. This is a two dimensional
nested list of ints >= 0.
observable (List[Union[str, List[List[List[float]]]]): A list with at least
one item and items are strings matching the observable regex or a two
dimensional hermitian matrix with complex entries. Each complex number is
represented using a List[float] of size 2, with element[0] being the real part
and element[1] imaginary. inf, -inf, and NaN are not allowable inputs for the element.
parameters (List[str]): The parameters used in the adjoint gradient calculation.
This is a list of parameter names. Default: all parameters.
Examples:
>>> AdjointGradient(
>>> targets=[[1, 2], [3]],
>>> observable=["2 * x @ y + z"],
>>> parameters=["theta", "beta"]
>>> )
"""
[docs]
class Type(str, Enum):
adjoint_gradient = "adjoint_gradient"
type = Type.adjoint_gradient
[docs]
class Sample(OptionalMultiTarget, Observable):
"""
Sample for specified targets and observable as requested result.
If no targets are specified, the observable must only operate on 1 qubit and it
will be applied to all qubits in parallel. Otherwise, the number of specified targets
must be equivalent to the number of qubits the observable can be applied to.
Attributes:
type (str): The result type. default = "sample". (type) is optional.
This should be unique among all result types.
targets (Optional[List[int]]): The target qubits. This is a list of int >= 0.
observable (List[Union[str, List[List[List[float]]]]): A list with at least
one item and items are strings matching the observable regex
or a two dimensional hermitian matrix with complex entries.
Each complex number is represented using a List[float] of size 2, with
element[0] being the real part and element[1] imaginary.
inf, -inf, and NaN are not allowable inputs for the element.
Examples:
>>> Sample(targets=[1], observable=["x"])
"""
[docs]
class Type(str, Enum):
sample = "sample"
type = Type.sample
[docs]
class Variance(OptionalMultiTarget, Observable):
"""
Variance of specified targets and observables as requested result.
If no targets are specified, the observable must only operate on 1 qubit and it
will be applied to all qubits in parallel. Otherwise, the number of specified targets
must be equivalent to the number of qubits the observable can be applied to.
Attributes:
type (str): The result type. default = "variance". (type) is optional.
This should be unique among all result types.
targets (List[int]): The target qubits. This is a list of int >= 0.
observable (List[Union[str, List[List[List[float]]]]): A list with at least
one item and items are strings matching the observable regex
or a two dimensional hermitian matrix with complex entries.
Each complex number is represented using a List[float] of size 2, with
element[0] being the real part and element[1] imaginary.
inf, -inf, and NaN are not allowable inputs for the element.
Examples:
>>> Variance(targets=[1], observable=["x"])
"""
[docs]
class Type(str, Enum):
variance = "variance"
type = Type.variance
[docs]
class StateVector(BaseModel):
"""
The full state vector as requested result.
Attributes:
type (str): The result type. default = "statevector". (type) is optional.
This should be unique among all result types.
Examples:
>>> StateVector()
"""
[docs]
class Type(str, Enum):
statevector = "statevector"
type = Type.statevector
[docs]
class DensityMatrix(OptionalMultiTarget):
"""
The density matrix as requested result.
Attributes:
type (str): The result type. default = "densitymatrix". (type) is optional.
This should be unique among all result types.
targets (Optional[List[int]]): The target qubits of the reduced density matrix.
This is a list of int >= 0.
Examples:
>>> DensityMatrix()
"""
[docs]
class Type(str, Enum):
densitymatrix = "densitymatrix"
type = Type.densitymatrix
[docs]
class Amplitude(MultiState):
"""
Amplitudes of specified states as requested result.
Attributes:
type (str): The result type. default = "amplitude". (type) is optional.
This should be unique among all result types.
states (List[string]): Variable length list with with all strings
matching the state regex
Examples:
>>> Amplitude(states=["01", "10"])
"""
[docs]
class Type(str, Enum):
amplitude = "amplitude"
type = Type.amplitude
[docs]
class Probability(OptionalMultiTarget):
"""
Probability of all states if no targets are specified or the marginal probability
of a restricted set of states if only a subset of all qubits are specified as targets
Attributes:
type (str): The result type. default = "probability". (type) is optional.
This should be unique among all result types.
targets (Optional[List[int]]): The target qubits. This is a list of int >= 0.
Examples:
>>> Probability(targets=[1, 2])
"""
[docs]
class Type(str, Enum):
probability = "probability"
type = Type.probability