diff --git a/test/unit_tests/braket/reservations/test_reservations.py b/test/unit_tests/braket/reservations/test_reservations.py index 07a1975b1..bbddaf4f7 100644 --- 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():