amazon-braket-sdk-python icon indicating copy to clipboard operation
amazon-braket-sdk-python copied to clipboard

feature: update device.run to support experimentalCapabilities flag

Open Altanali opened this issue 1 month ago • 1 comments

Issue #, if available:

Description of changes:

Updating AwsQuantumTask and device.run(...) to support the new experimentalCapabilities argument in Braket:CreateQuantumTask. We introduce the enabled_experimental_capabilities argument which currently only accepts `["ALL"]`` as a valid input and adds

{ 
 "experimentalCapabilities:: {
    "enabled": "ALL"
    }
}

to the quantum task arguments.

Additionally, update the EnableExperimentalCapabilities context manager to support taking in a list of experimental capabilities.

Testing done:

  • Added unittests for the context manager.
  • Updated unit tests for AwsDevice, AwsQuantumTask, AwsQuantumTaskBatch, and test utils to support the enabled_experimental_capabilities argument.

Merge Checklist

Put an x in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.

General

Tests

  • [x] I have added tests that prove my fix is effective or that my feature works (if appropriate)
  • [x] I have checked that my tests are not configured for a specific region or account (if appropriate)

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

Altanali avatar Dec 15 '25 19:12 Altanali

Codecov Report

:white_check_mark: All modified and coverable lines are covered by tests. :white_check_mark: Project coverage is 100.00%. Comparing base (bdc945c) to head (04755d1).

Additional details and impacted files
@@            Coverage Diff            @@
##              main     #1183   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files          162       162           
  Lines        10504     10540   +36     
  Branches      1308      1316    +8     
=========================================
+ Hits         10504     10540   +36     

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.

:rocket: New features to boost your workflow:
  • :snowflake: Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

codecov[bot] avatar Dec 16 '25 00:12 codecov[bot]