This is an automated email from the ASF dual-hosted git repository.
guanmingchiu pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/mahout.git
The following commit(s) were added to refs/heads/main by this push:
new 0c7ed0e32 feat: Add AWS session support for backend initialization in
Amazon Braket (#976)
0c7ed0e32 is described below
commit 0c7ed0e3205133b621d28354bc74dbae7daf1451
Author: Ryan Huang <[email protected]>
AuthorDate: Thu Jan 29 20:56:54 2026 +0800
feat: Add AWS session support for backend initialization in Amazon Braket
(#976)
---
qumat/amazon_braket_backend.py | 21 ++++++++++++++++++---
1 file changed, 18 insertions(+), 3 deletions(-)
diff --git a/qumat/amazon_braket_backend.py b/qumat/amazon_braket_backend.py
index 8710ab0f9..6b4061a7b 100644
--- a/qumat/amazon_braket_backend.py
+++ b/qumat/amazon_braket_backend.py
@@ -14,7 +14,8 @@
# See the License for the specific language governing permissions and
# limitations under the License.
#
-from braket.aws import AwsDevice
+import boto3
+from braket.aws import AwsDevice, AwsSession
from braket.devices import LocalSimulator
from braket.circuits import Circuit, FreeParameter
@@ -22,15 +23,29 @@ from braket.circuits import Circuit, FreeParameter
def initialize_backend(backend_config):
backend_options = backend_config["backend_options"]
simulator_type = backend_options.get("simulator_type", "default")
+ region = backend_options.get("region")
+
+ # Create AWS session with region if specified
+ aws_session = None
+ if region:
+ boto_session = boto3.Session(region_name=region)
+ aws_session = AwsSession(boto_session=boto_session)
+
if simulator_type == "local":
return LocalSimulator()
elif simulator_type == "default":
- return
AwsDevice("arn:aws:braket:::device/quantum-simulator/amazon/sv1")
+ return AwsDevice(
+ "arn:aws:braket:::device/quantum-simulator/amazon/sv1",
+ aws_session=aws_session,
+ )
else:
print(
f"Simulator type '{simulator_type}' is not supported in Amazon
Braket. Using default."
)
- return
AwsDevice("arn:aws:braket:::device/quantum-simulator/amazon/sv1")
+ return AwsDevice(
+ "arn:aws:braket:::device/quantum-simulator/amazon/sv1",
+ aws_session=aws_session,
+ )
def create_empty_circuit(num_qubits: int | None = None):
