fix: tests

amazon-braket · Apr 16, 2024 · 4173a94 · 4173a94
1 parent df50f40
commit 4173a94
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/unit_tests/braket/reservations/test_reservations.py b/test/unit_tests/braket/reservations/test_reservations.py
@@ -22,8 +22,9 @@
 
 @pytest.fixture
 def aws_device():
-    device = AwsDevice("device_arn_example")
-    return device
+    mock_device = MagicMock(name="AwsDevice")
+    mock_device.device_arn = "device_arn_example"
+    return mock_device
 
 
 def test_initialization_with_string():