Using device wires analytic_probability method (#57)

* Using device wires instead of ascending scalar indices * Increasing the number of shots
PennyLaneAI · Aug 10, 2020 · 48c150a · 48c150a
1 parent c1efdad
commit 48c150a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pennylane_forest/device.py b/pennylane_forest/device.py
@@ -361,7 +361,7 @@ def analytic_probability(self, wires=None):
         if self._state is None:
             return None
 
-        wires = wires or range(self.num_wires)
+        wires = wires or self.wires
         wires = Wires(wires)
         prob = self.marginal_prob(np.abs(self._state) ** 2, wires)
         return prob
diff --git a/tests/test_qpu.py b/tests/test_qpu.py
@@ -118,14 +118,14 @@ def test_tensor_expval_operator_estimation(self, obs):
         dev = qml.device(
             "forest.qpu",
             device=device,
-            shots=1000,
+            shots=2000,
             load_qc=False,
             parametric_compilation=False,
         )
         dev_1 = qml.device(
             "forest.qpu",
             device=device,
-            shots=1000,
+            shots=2000,
             load_qc=False,
             parametric_compilation=False,
         )