Skip to content

Commit 9424125

Browse files
simplify conditional
1 parent 3d0f22f commit 9424125

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

sentry_sdk/integrations/boto3.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def _sentry_request_created(
114114
def _sentry_after_call(
115115
context: "Dict[str, Any]", parsed: "Dict[str, Any]", **kwargs: "Any"
116116
) -> None:
117-
span: "Optional[Span]" = context.pop("_sentrysdk_span", None)
117+
span: "Optional[Union[Span, StreamedSpan]]" = context.pop("_sentrysdk_span", None)
118118

119119
# Span could be absent if the integration is disabled.
120120
if span is None:
@@ -125,11 +125,7 @@ def _sentry_after_call(
125125
if not isinstance(body, StreamingBody):
126126
return
127127

128-
client = sentry_sdk.get_client()
129-
span_streaming = has_span_streaming_enabled(client.options)
130-
131-
streaming_span: "Union[Span, StreamedSpan]"
132-
if span_streaming:
128+
if isinstance(span, StreamedSpan):
133129
streaming_span = sentry_sdk.traces.start_span(
134130
name=span.name,
135131
parent_span=span,

0 commit comments

Comments
 (0)