fix tests

PennyLaneAI · Dec 2, 2024 · 5620230 · 5620230
1 parent 588c72a
commit 5620230
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_device.py b/tests/test_device.py
@@ -468,7 +468,7 @@ def __copy__(self):
     def test_default_noise_model(self, requires_api):
         """Test that the default noise model is correctly set."""
         dev = SimulatorDevice(wires=(0,))
-        assert dev.noise_model == "ideal"
+        assert dev.job["input"]["noise"]["model"] == "ideal"
 
 class TestJobAttribute:
     """Tests job creation with mocked submission."""
@@ -731,4 +731,4 @@ def mock_submit_job(*args):
 
         dev.apply(tape.operations)
 
-        assert dev.job["noise"]["model"] == "forte-1"
+        assert dev.job["input"]["noise"]["model"] == "forte-1"