From 27bd6a9b1768a5d56acaf7df3151fbc8c1e86358 Mon Sep 17 00:00:00 2001
From: Cody Wang <speller26@gmail.com>
Date: Mon, 28 Oct 2024 12:39:24 -0700
Subject: [PATCH] Update src/braket/pennylane_plugin/translation.py

Co-authored-by: Christina Lee <chrissie.c.l@gmail.com>
---
 src/braket/pennylane_plugin/translation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)