chore: Apply code style changes

Author: yiyuan-he

aws-opentelemetry-distro/src/amazon/opentelemetry/distro/patches/_botocore_patches.py

@@ -51,7 +51,7 @@ def _apply_botocore_lambda_patch() -> None:
 
     This patch adds an extension to the upstream's list of known extensions for Lambda.
     Extensions allow for custom logic for adding service-specific information to spans,
-    such as attributes. Specifically, we are adding logic to add the 
+    such as attributes. Specifically, we are adding logic to add the
     `aws.lambda.function.name` and  `aws.lambda.resource_mapping.id` attributes
 
     Sidenote: There exists SpanAttributes.FAAS_INVOKED_NAME for invoke operations
@@ -70,7 +70,7 @@ def patch_extract_attributes(self, attributes: _AttributeMapT):
         if resource_mapping_id:
             attributes[AWS_LAMBDA_RESOURCEMAPPING_ID] = resource_mapping_id
 
-    old_on_success  = _LambdaExtension.on_success 
+    old_on_success = _LambdaExtension.on_success
 
     def patch_on_success(self, span: Span, result: _BotoResultT):
         old_on_success(self, span, result)
@@ -86,7 +86,7 @@ def patch_on_success(self, span: Span, result: _BotoResultT):
 def _apply_botocore_stepfunctions_patch() -> None:
     """Botocore instrumentation patch for StepFunctions
 
-    This patch adds an extension to the upstream's list of known extensions for 
+    This patch adds an extension to the upstream's list of known extensions for
     StepFunctions. Extensions allow for custom logic for adding service-specific
     information to spans, such as attributes. Specifically, we are adding logic
     to add the `aws.stepfunctions.state_machine.arn` and `aws.stepfunctions.activity.arn`
@@ -101,7 +101,7 @@ def _apply_botocore_sns_patch() -> None:
 
     This patch adds an extension to the upstream's list of known extensions for SNS.
     Extensions allow for custom logic for adding service-specific information to
-    spans, such as attributes. Specifically, we are adding logic to add the 
+    spans, such as attributes. Specifically, we are adding logic to add the
     `aws.sns.topic.arn` attribute, to be used to generate RemoteTarget and achieve
     parity with the Java instrumentation.
 
@@ -119,12 +119,13 @@ def patch_extract_attributes(self, attributes: _AttributeMapT):
 
     _SnsExtension.extract_attributes = patch_extract_attributes
 
+
 def _apply_botocore_secretsmanager_patch() -> None:
     """Botocore instrumentation patch for SecretsManager
 
     This patch adds an extension to the upstream's list of known extension for SecretsManager.
     Extensions allow for custom logic for adding service-specific information to spans, such as
-    attributes. Specifically, we are adding logic to add the `aws.secretsmanager.secret.arn` 
+    attributes. Specifically, we are adding logic to add the `aws.secretsmanager.secret.arn`
     attribute, to be used to generate RemoteTarget and achieve parity with the Java
     instrumentation.
     """

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test_aws_metric_attribute_generator.py

@@ -1108,7 +1108,7 @@ def test_sdk_client_span_with_remote_resource_attributes(self):
             [AWS_SECRETSMANAGER_SECRET_ARN],
             ["arn:aws:secretsmanager:us-east-1:123456789012:secret:secret_name-lERW9H"],
             keys,
-            values
+            values,
         )
         self._validate_remote_resource_attributes(
             "AWS::SecretsManager::Secret", "arn:aws:secretsmanager:us-east-1:123456789012:secret:secret_name-lERW9H"
@@ -1122,14 +1122,13 @@ def test_sdk_client_span_with_remote_resource_attributes(self):
 
         # Validate behaviour of AWS_STEPFUNCTIONS_STATEMACHINE_ARN attribute, then remove it.
         self._mock_attribute(
-            [AWS_STEPFUNCTIONS_STATEMACHINE_ARN], 
-            ["arn:aws:states:us-east-1:123456789012:stateMachine:test_state_machine"], 
-             keys, 
-             values
+            [AWS_STEPFUNCTIONS_STATEMACHINE_ARN],
+            ["arn:aws:states:us-east-1:123456789012:stateMachine:test_state_machine"],
+            keys,
+            values,
         )
         self._validate_remote_resource_attributes(
-            "AWS::StepFunctions::StateMachine",
-            "arn:aws:states:us-east-1:123456789012:stateMachine:test_state_machine"
+            "AWS::StepFunctions::StateMachine", "arn:aws:states:us-east-1:123456789012:stateMachine:test_state_machine"
         )
         self._mock_attribute([AWS_STEPFUNCTIONS_STATEMACHINE_ARN], [None])
 
@@ -1152,10 +1151,7 @@ def test_sdk_client_span_with_remote_resource_attributes(self):
             keys,
             values,
         )
-        self._validate_remote_resource_attributes(
-            "AWS::Kinesis::StreamConsumer", 
-            "aws_stream_consumername"
-        )
+        self._validate_remote_resource_attributes("AWS::Kinesis::StreamConsumer", "aws_stream_consumername")
         self._mock_attribute([AWS_KINESIS_STREAM_CONSUMERNAME], [None])
 
         # Validate behaviour with both AWS_KINESIS_STREAM_NAME and AWS_KINESIS_STREAM_CONSUMERNAME
@@ -1185,7 +1181,7 @@ def test_sdk_client_span_with_remote_resource_attributes(self):
             [AWS_LAMBDA_FUNCTION_NAME, AWS_LAMBDA_RESOURCEMAPPING_ID],
             ["aws_lambda_function_name", "aws_event_source_mapping_id"],
             keys,
-            values
+            values,
         )
         self._validate_remote_resource_attributes("AWS::Lambda::EventSourceMapping", "aws_event_source_mapping_id")
         self._mock_attribute([AWS_LAMBDA_FUNCTION_NAME, AWS_LAMBDA_RESOURCEMAPPING_ID], [None, None])