diff --git a/src/braket/pennylane_plugin/translation.py b/src/braket/pennylane_plugin/translation.py index 76c34622..d8cad22c 100644 --- a/src/braket/pennylane_plugin/translation.py +++ b/src/braket/pennylane_plugin/translation.py @@ -576,7 +576,7 @@ def translate_result_type( # noqa: C901 observable = measurement.obs if return_type is ObservableReturnTypes.Probability: - if observable: + if observable and observable.diagonalizing_gates(): raise qml.DeviceError("Probability result type not supported for observables") return Probability(targets)