Fix minor bugs

qiboteam · Dec 12, 2023 · 865a0ba · 865a0ba
1 parent 6b45d25
commit 865a0ba
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/qibochem/measurement/expectation.py b/src/qibochem/measurement/expectation.py
@@ -1,5 +1,6 @@
 from functools import reduce
 
+import numpy as np
 import qibo
 from qibo import gates
 from qibo.hamiltonians import SymbolicHamiltonian
@@ -73,6 +74,7 @@ def expectation(
         if n_shots_per_pauli_term:
             total = sum(pauli_term_sample_expectation(circuit, term, n_shots) for term in hamiltonian.terms)
         else:
+            n_terms = len(hamiltonian.terms)
             # Determine how to allocate n_shots first
             if shot_distribution == "uniform":
                 # Split evenly amongst all the terms in the Hamiltonian