Fix import in shots=None case

unitaryfund · Oct 21, 2024 · 034ec88 · 034ec88
1 parent cc7a844
commit 034ec88
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pennylane_qrack/qrack_device.py b/pennylane_qrack/qrack_device.py
@@ -24,7 +24,7 @@
 
 import numpy as np
 
-from pennylane import QubitDevice, DeviceError
+from pennylane import QubitDevice, DeviceError, QuantumFunctionError
 from pennylane.ops import (
     QubitStateVector,
     BasisState,
@@ -649,7 +649,7 @@ def expval(self, observable, **kwargs):
 
     def generate_samples(self):
         if self.shots is None:
-            raise qml.QuantumFunctionError(
+            raise QuantumFunctionError(
                 "The number of shots has to be explicitly set on the device "
                 "when using sample-based measurements."
             )