braket.jobs.image_uris module

class braket.jobs.image_uris.Framework(value)[source]

Bases: str, Enum

Supported Frameworks for pre-built containers

BASE = 'BASE'
PL_TENSORFLOW = 'PL_TENSORFLOW'
PL_PYTORCH = 'PL_PYTORCH'
braket.jobs.image_uris.built_in_images(region: str) → set[str][source]

Checks a region for built in Braket images.

Parameters:

region (str) – The AWS region to check for images

Returns:

set[str] – returns a set of built images

braket.jobs.image_uris.retrieve_image(framework: Framework, region: str) → str[source]

Retrieves the ECR URI for the Docker image matching the specified arguments.

Parameters:

  • framework (Framework) – The name of the framework.

  • region (str) – The AWS region for the Docker image.

Returns:

str – The ECR URI for the corresponding Amazon Braket Docker image.

Raises:

ValueError – If any of the supplied values are invalid or the combination of inputs specified is not supported.