Skip to content

Commit 6a135d9

Browse files
authored
Use Self type for context manager in telemetry.py
1 parent 210e93b commit 6a135d9

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/a2a/utils/telemetry.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ def internal_method(self):
7171

7272
from collections.abc import Callable
7373
from typing import TYPE_CHECKING, Any
74+
from typing_extensions import Self
7475

7576

7677
if TYPE_CHECKING:
@@ -119,7 +120,7 @@ class _NoOp:
119120
def __call__(self, *args: Any, **kwargs: Any) -> Any:
120121
return self
121122

122-
def __enter__(self) -> '_NoOp':
123+
def __enter__(self) -> Self:
123124
return self
124125

125126
def __exit__(self, *args: object, **kwargs: Any) -> None:

0 commit comments

Comments
 (0)