feat: env var method

.pre-commit-config.yaml

@@ -0,0 +1,8 @@
+repos:
+  - repo: local
+    hooks:
+      - id: tox_linters
+        name: linters
+        entry: tox -e linters_check
+        language: system
+        types: [python]

examples/reservation.py

@@ -14,26 +14,18 @@
 from braket.aws import AwsDevice
 from braket.circuits import Circuit
 from braket.devices import Devices
-from braket.reservations import reservation
+from braket.reservations import DirectReservation
 
 bell = Circuit().h(0).cnot(0, 1)
 device = AwsDevice(Devices.IonQ.Aria1)
 
 # To run a task in a device reservation, change the device to the one you reserved
 # and fill in your reservation ARN.
-with reservation(device, reservation_arn="reservation ARN"):
+with DirectReservation(device, reservation_arn="reservation ARN"):
     task = device.run(bell, shots=100)
-
 print(task.result().measurement_counts)
 
-# Alternatively, you may directly pass the reservation ARN to all quantum tasks.
+# Alternatively, you may start the reservation globally
+DirectReservation(device, reservation_arn="reservation ARN").start()
 task = device.run(bell, shots=100, reservation_arn="reservation ARN")
 print(task.result().measurement_counts)
-
-
-# A third option to run a reservation is by decorating a function.
-# All tasks created within the function will use the reservation.
-@reservation(device, reservation_arn="reservation ARN")
-def my_function():
-    task = device.run(bell, shots=100)
-    print(task.result().measurement_counts)

src/braket/reservations/__init__.py

@@ -12,4 +12,4 @@
 # language governing permissions and limitations under the License.
 
 
-from braket.reservations.reservations import reservation  # noqa: F401
+from braket.reservations.reservations import DirectReservation  # noqa: F401

test/unit_tests/braket/aws/test_aws_session.py

@@ -1415,10 +1415,8 @@ def test_create_quantum_task_with_correct_device_and_reservation():
 
     # Expected boto3 parameters
     boto3_params = {"deviceArn": "device_arn_example"}
-
     expected_arn = "example_quantum_task_arn"
 
-    # Mock boto3 client and its response
     with patch("boto3.client") as mock_boto3_client:
         mock_client = MagicMock()
         mock_client.create_quantum_task.return_value = {"quantumTaskArn": expected_arn}
@@ -1430,7 +1428,6 @@ def test_create_quantum_task_with_correct_device_and_reservation():
         # Assertions
         mock_client.create_quantum_task.assert_called_once_with(
             deviceArn="device_arn_example",
-            someOtherParam="value",
             reservationArn="reservation_arn_example",
         )
         assert arn == expected_arn