Fix fom tests

Shaobo-Zhou · Shaobo-Zhou · commit 22c044ebf72b · 2026-02-15T11:54:20.000+01:00
diff --git a/tests/compilation/test_predictor_rl.py b/tests/compilation/test_predictor_rl.py
@@ -147,7 +147,7 @@ def test_fom_aware_compile_fallback(monkeypatch: MonkeyPatch) -> None:
     monkeypatch.setattr(
         predictor.env, "calculate_reward", lambda _circ: (_ for _ in ()).throw(RuntimeError("fake error"))
     )
-    compiled_qc, prop_set = predictor.env.fom_aware_compile(dummy_action, None, qc, max_iteration=1)
+    compiled_qc, prop_set = predictor.env.fom_aware_compile(dummy_action, predictor.env.device, qc, max_iteration=1)
 
     assert isinstance(compiled_qc, QuantumCircuit)
     assert isinstance(prop_set, dict)

Original file line number	Diff line number	Diff line change
`@@ -147,7 +147,7 @@ def test_fom_aware_compile_fallback(monkeypatch: MonkeyPatch) -> None:`
`147`	`147`	`monkeypatch.setattr(`
`148`	`148`	`predictor.env, "calculate_reward", lambda _circ: (_ for _ in ()).throw(RuntimeError("fake error"))`
`149`	`149`	`)`
`150`		`- compiled_qc, prop_set = predictor.env.fom_aware_compile(dummy_action, None, qc, max_iteration=1)`
	`150`	`+ compiled_qc, prop_set = predictor.env.fom_aware_compile(dummy_action, predictor.env.device, qc, max_iteration=1)`
`151`	`151`
`152`	`152`	`assert isinstance(compiled_qc, QuantumCircuit)`
`153`	`153`	`assert isinstance(prop_set, dict)`