More typing

Author: xrmx

opentelemetry-api/src/opentelemetry/_logs/_internal/__init__.py

@@ -37,10 +37,9 @@
 from logging import getLogger
 from os import environ
 from time import time_ns
-from typing import Optional, Union, cast
+from typing import Optional, cast
 
 from opentelemetry._logs.severity import SeverityNumber
-from opentelemetry.attributes import BoundedAttributes
 from opentelemetry.environment_variables import _OTEL_PYTHON_LOGGER_PROVIDER
 from opentelemetry.trace.span import TraceFlags
 from opentelemetry.util._once import Once
@@ -68,9 +67,7 @@ def __init__(
         severity_text: Optional[str] = None,
         severity_number: Optional[SeverityNumber] = None,
         body: AnyValue = None,
-        attributes: Optional[
-            Union[BoundedAttributes, ExtendedAttributes]
-        ] = None,
+        attributes: Optional[ExtendedAttributes] = None,
     ):
         self.timestamp = timestamp
         if observed_timestamp is None:

opentelemetry-sdk/src/opentelemetry/sdk/_events/__init__.py

@@ -21,7 +21,7 @@
 from opentelemetry._events import EventLoggerProvider as APIEventLoggerProvider
 from opentelemetry._logs import NoOpLogger, SeverityNumber, get_logger_provider
 from opentelemetry.sdk._logs import Logger, LoggerProvider, LogRecord
-from opentelemetry.util.types import Attributes
+from opentelemetry.util.types import ExtendedAttributes
 
 _logger = logging.getLogger(__name__)
 
@@ -33,7 +33,7 @@ def __init__(
         name: str,
         version: Optional[str] = None,
         schema_url: Optional[str] = None,
-        attributes: Optional[Attributes] = None,
+        attributes: Optional[ExtendedAttributes] = None,
     ):
         super().__init__(
             name=name,
@@ -74,7 +74,7 @@ def get_event_logger(
         name: str,
         version: Optional[str] = None,
         schema_url: Optional[str] = None,
-        attributes: Optional[Attributes] = None,
+        attributes: Optional[ExtendedAttributes] = None,
     ) -> EventLogger:
         if not name:
             _logger.warning("EventLogger created with invalid name: %s", name)

opentelemetry-sdk/src/opentelemetry/sdk/_logs/_internal/__init__.py

@@ -24,7 +24,7 @@
 from os import environ
 from threading import Lock
 from time import time_ns
-from typing import Any, Callable, Tuple, Union  # noqa
+from typing import Any, Callable, Tuple, Union, cast  # noqa
 
 from opentelemetry._logs import Logger as APILogger
 from opentelemetry._logs import LoggerProvider as APILoggerProvider
@@ -251,8 +251,11 @@ def to_json(self, indent: int | None = 4) -> str:
 
     @property
     def dropped_attributes(self) -> int:
-        if self.attributes:
-            return self.attributes.dropped
+        attributes: BoundedAttributes = cast(
+            BoundedAttributes, self.attributes
+        )
+        if attributes:
+            return attributes.dropped
         return 0