Update src/braket/pennylane_plugin/translation.py

Co-authored-by: Christina Lee <[email protected]>
amazon-braket · Oct 28, 2024 · 27bd6a9 · 27bd6a9
1 parent 5ca54ff
commit 27bd6a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)