fixing mypy issues

Author: mrm9084

featuremanagement/azuremonitor/_send_telemetry.py

@@ -10,7 +10,9 @@
 try:
     from azure.monitor.events.extension import track_event as azure_monitor_track_event  # type: ignore
     from opentelemetry import trace, baggage, context
+    from opentelemetry.context.context import Context
     from opentelemetry.sdk.trace import Span, SpanProcessor
+    from opentelemetry.util import types
 
     HAS_AZURE_MONITOR_EVENTS_EXTENSION = True
 except ImportError:
@@ -110,7 +112,7 @@ def publish_telemetry(evaluation_event: EvaluationEvent) -> None:
     track_event(EVENT_NAME, evaluation_event.user, event_properties=event)
 
 
-def attach_targeting_info(targeting_id: str):
+def attach_targeting_info(targeting_id: str) -> None:
     """
     Attaches the targeting ID to the current span and baggage.
 
@@ -125,13 +127,13 @@ class TargetingSpanProcessor(SpanProcessor):
     A custom SpanProcessor that attaches the targeting ID to the span and baggage when a new span is started.
     """
 
-    def on_start(self, span: Span, parent_context=None):
+    def on_start(self, span: Span, parent_context: Optional[Context] =None) -> None:
         """
         Attaches the targeting ID to the span and baggage when a new span is started.
 
         :param Span span: The span that was started.
         :param parent_context: The parent context of the span.
         """
         target_baggage = baggage.get_baggage(MICROSOFT_TARGETING_ID, parent_context)
-        if target_baggage is not None:
+        if target_baggage is not None and isinstance(target_baggage, str):
             span.set_attribute(TARGETING_ID, target_baggage)

tests/requirements.txt

@@ -1,2 +1,2 @@
 azure-monitor-opentelemetry
-azure-monitor-events-extension
+azure-monitor-events-extension