def __init__()

in src/braket/aws/aws_session.py [0:0]

• 28 lines of code
• 8 McCabe index (conditional complexity)

    def __init__(self, boto_session=None, braket_client=None, config=None, default_bucket=None):
        """
        Args:
            boto_session: A boto3 session object.
            braket_client: A boto3 Braket client.
            config: A botocore Config object.
        """
        if (
            boto_session
            and braket_client
            and boto_session.region_name != braket_client.meta.region_name
        ):
            raise ValueError(
                "Boto Session region and Braket Client region must match and currently "
                f"they do not: Boto Session region is '{boto_session.region_name}', but "
                f"Braket Client region is '{braket_client.meta.region_name}'."
            )

        self._config = config

        if braket_client:
            self.boto_session = boto_session or boto3.Session(
                region_name=braket_client.meta.region_name
            )
            self.braket_client = braket_client
        else:
            self.boto_session = boto_session or boto3.Session()
            self.braket_client = self.boto_session.client("braket", config=self._config)

        self._update_user_agent()
        self._custom_default_bucket = bool(default_bucket)
        self._default_bucket = default_bucket or os.environ.get("AMZN_BRAKET_OUT_S3_BUCKET")

        self._iam = None
        self._s3 = None
        self._sts = None
        self._logs = None
        self._ecr = None