Fix failing CI after Pandas 2.0 update (#6051)

quantumlib · Apr 3, 2023 · 6b0c098 · 6b0c098
1 parent f114996
commit 6b0c098
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cirq-core/cirq/experiments/xeb_simulation_test.py b/cirq-core/cirq/experiments/xeb_simulation_test.py
@@ -59,7 +59,7 @@ def test_simulate_circuit_length_validation():
         )
         for _ in range(2)
     ]
-    cycle_depths = np.arange(3, 50, 9)
+    cycle_depths = np.arange(3, 50, 9, dtype=np.int64)
     with pytest.raises(ValueError, match='.*not long enough.*'):
         _ = simulate_2q_xeb_circuits(circuits=circuits, cycle_depths=cycle_depths)
 
@@ -127,7 +127,7 @@ def test_incremental_simulate(multiprocess):
         )
         for _ in range(20)
     ]
-    cycle_depths = np.arange(3, 100, 9)
+    cycle_depths = np.arange(3, 100, 9, dtype=np.int64)
 
     if multiprocess:
         pool = multiprocessing.Pool()