Make minor but critical update to baseline truncation numbers

stanfordLINQS · Jul 20, 2024 · d452e62 · d452e62
1 parent 66ec2e9
commit d452e62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/SQcircuit/circuit.py b/SQcircuit/circuit.py
@@ -1858,12 +1858,12 @@ def truncate_circuit(self, K: int, heuristic=False) -> List[int]:
             for mode in self.omega:
                 # charge mode
                 if mode == 0:
-                    trunc_nums.append(np.ceil(trunc_num_average))
+                    trunc_nums.append(int(np.ceil(trunc_num_average)))
                 else:
                     h = (A * trunc_num_average) / mode
                     trunc_nums.append(np.ceil(h))
         else:
-            trunc_nums = [np.ceil(trunc_num_average) for _ in range(len(self.omega))]
+            trunc_nums = [int(np.ceil(trunc_num_average)) for _ in range(len(self.omega))]
 
         self.set_trunc_nums(trunc_nums)
         return trunc_nums