From 4cf55222309140a66adef381358a9b18e59f030b Mon Sep 17 00:00:00 2001 From: hmunozb Date: Wed, 18 Sep 2024 17:37:28 -0700 Subject: [PATCH] fix tests --- tests/test_tts.py | 2 +- tests/tests_qubo_parallel.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_tts.py b/tests/test_tts.py index 53926f0..fefec53 100644 --- 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 index 5b28b95..c07ba5b 100644 --- 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)