fix tests

nasa · Sep 19, 2024 · 4cf5522 · 4cf5522
1 parent 3378094
commit 4cf5522
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_tts.py b/tests/test_tts.py
@@ -291,7 +291,7 @@ def test_tts(n_vars: int):
                                   30,
                                   1.0,
                                   3.5,
-                                  n_repetitions=201)
+                                  n_repetitions=101)
 
     bench_tts = raw_to_tts(bench, gap_key='energy_gap', ptgt=0.9)
     bench_tts_samps = tts_boots(bench_tts, 50, 0.9)

diff --git a/tests/tests_qubo_parallel.py b/tests/tests_qubo_parallel.py
@@ -268,7 +268,7 @@ def test_sequential_sweep_simulation_qubo(n_vars: int):
 
     # Fix temperature
     betas = np.array([10], dtype=dtype)
-    beta = np.arange(1)
+    beta_idx = np.arange(1)
     # Get initial state
     states = np.random.randint(2, size=(1, n_vars)).astype(dtype)