return type annotation

Author: alexander-alderman-webb

sentry_sdk/integrations/rust_tracing.py

@@ -184,7 +184,9 @@ def on_event(self, event: str, sentry_span: "SentrySpan") -> None:
         elif event_type == EventTypeMapping.Event:
             process_event(deserialized_event)
 
-    def on_new_span(self, attrs: str, span_id: str) -> "Optional[SentrySpan]":
+    def on_new_span(
+        self, attrs: str, span_id: str
+    ) -> "Optional[Union[SentrySpan, StreamedSpan]]":
         attrs = json.loads(attrs)
         metadata = attrs.get("metadata", {})