Fix.

aws-observability · May 30, 2024 · e07ea8d · e07ea8d
1 parent a0b05f5
commit e07ea8d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/contract-tests/tests/test/amazon/botocore/botocore_test.py b/contract-tests/tests/test/amazon/botocore/botocore_test.py
@@ -20,7 +20,7 @@
 )
 from opentelemetry.proto.common.v1.common_pb2 import AnyValue, KeyValue
 from opentelemetry.proto.metrics.v1.metrics_pb2 import ExponentialHistogramDataPoint, Metric
-from opentelemetry.proto.trace.v1.trace_pb2 import Span, SpanKind
+from opentelemetry.proto.trace.v1.trace_pb2 import Span
 from opentelemetry.semconv.trace import SpanAttributes
 
 _logger: Logger = getLogger(__name__)
@@ -467,7 +467,9 @@ def _assert_aws_span_attributes(self, resource_scope_spans: List[ResourceScopeSp
         for resource_scope_span in resource_scope_spans:
             # pylint: disable=no-member
             if resource_scope_span.span.kind in (Span.SPAN_KIND_CLIENT, Span.SPAN_KIND_PRODUCER):
-                self.assertEqual(resource_scope_span.span.kind, Span.SpanKind.Value('SPAN_KIND_' + kwargs.get("span_kind")))
+                self.assertEqual(
+                    resource_scope_span.span.kind, Span.SpanKind.Value("SPAN_KIND_" + kwargs.get("span_kind"))
+                )
                 target_spans.append(resource_scope_span.span)
         self.assertEqual(len(target_spans), 1)
         self._assert_aws_attributes(
@@ -510,7 +512,9 @@ def _assert_semantic_conventions_span_attributes(
         for resource_scope_span in resource_scope_spans:
             # pylint: disable=no-member
             if resource_scope_span.span.kind in (Span.SPAN_KIND_CLIENT, Span.SPAN_KIND_PRODUCER):
-                self.assertEqual(resource_scope_span.span.kind, Span.SpanKind.Value('SPAN_KIND_' + kwargs.get("span_kind")))
+                self.assertEqual(
+                    resource_scope_span.span.kind, Span.SpanKind.Value("SPAN_KIND_" + kwargs.get("span_kind"))
+                )
                 target_spans.append(resource_scope_span.span)
 
         self.assertEqual(len(target_spans), 1)