BP sample tests: fix seed

jcmgray · Oct 20, 2023 · 4b8d35f · 4b8d35f
1 parent 91c4266
commit 4b8d35f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_tensor/test_belief_propagation/test_hd1bp.py b/tests/test_tensor/test_belief_propagation/test_hd1bp.py
@@ -34,7 +34,7 @@ def test_contract_normal(damping):
 def test_sample(damping):
     nvars = 20
     htn = qtn.HTN_random_ksat(3, nvars, alpha=2.0, seed=42, mode="dense")
-    config, tn_config, omega = sample_hd1bp(htn, damping=damping)
+    config, tn_config, omega = sample_hd1bp(htn, damping=damping, seed=42)
     assert len(config) == nvars
     assert tn_config.num_indices == 0
     assert tn_config.contract() == pytest.approx(1.0)

diff --git a/tests/test_tensor/test_belief_propagation/test_hv1bp.py b/tests/test_tensor/test_belief_propagation/test_hv1bp.py
@@ -34,7 +34,7 @@ def test_contract_normal(damping):
 def test_sample(damping):
     nvars = 20
     htn = qtn.HTN_random_ksat(3, nvars, alpha=2.0, seed=42, mode="dense")
-    config, tn_config, omega = sample_hv1bp(htn, damping=damping)
+    config, tn_config, omega = sample_hv1bp(htn, damping=damping, seed=42)
     assert len(config) == nvars
     assert tn_config.num_indices == 0
     assert tn_config.contract() == pytest.approx(1.0)