chore(profiling): fix typing issues in stack_v2 (#15119)

## Description

This updates the stubs in Stack V2 to fix some typing issues we had.

Author: KowalskiThomas

ddtrace/internal/datadog/profiling/stack_v2/__init__.pyi

@@ -1,9 +1,20 @@
 import asyncio
+from types import FrameType
+from typing import Optional, Sequence, Union
 
 def register_thread(id: int, native_id: int, name: str) -> None: ...  # noqa: A002
 def unregister_thread(name: str) -> None: ...
 def track_asyncio_loop(thread_id: int, loop: asyncio.AbstractEventLoop) -> None: ...
 def link_tasks(parent: asyncio.AbstractEventLoop, child: asyncio.Task) -> None: ...
+def init_asyncio(
+    current_tasks: Sequence[asyncio.Task],
+    scheduled_tasks: Sequence[asyncio.Task],
+    eager_tasks: Optional[Sequence[asyncio.Task]],
+) -> None: ...
+def track_greenlet(greenlet_id: int, name: str, frame: Union[FrameType, bool, None]) -> None: ...
+def untrack_greenlet(greenlet_id: int) -> None: ...
+def link_greenlets(greenlet_id: int, parent_id: int) -> None: ...
+def update_greenlet_frame(greenlet_id: int, frame: Union[FrameType, bool, None]) -> None: ...
 
 is_available: bool
 failure_msg: str

ddtrace/profiling/_asyncio.py

@@ -98,7 +98,7 @@ def _(f, args, kwargs):
             scheduled_tasks = asyncio.tasks._all_tasks.data
             eager_tasks = None
 
-        stack_v2.init_asyncio(asyncio.tasks._current_tasks, scheduled_tasks, eager_tasks)  # type: ignore[attr-defined]
+        stack_v2.init_asyncio(asyncio.tasks._current_tasks, scheduled_tasks, eager_tasks)
 
 
 def get_event_loop_for_thread(thread_id: int) -> typing.Union["asyncio.AbstractEventLoop", None]:

ddtrace/profiling/_gevent.py

@@ -38,7 +38,7 @@ def track_gevent_greenlet(greenlet: _Greenlet) -> _Greenlet:
     frame: t.Union[FrameType, bool, None] = FRAME_NOT_SET
 
     try:
-        stack_v2.track_greenlet(greenlet_id, greenlet.name or type(greenlet).__qualname__, frame)  # type: ignore[attr-defined]
+        stack_v2.track_greenlet(greenlet_id, greenlet.name or type(greenlet).__qualname__, frame)
     except AttributeError as e:
         raise GreenletTrackingError("Cannot track greenlet with no name attribute") from e
     except Exception as e:
@@ -60,7 +60,7 @@ def track_gevent_greenlet(greenlet: _Greenlet) -> _Greenlet:
 
 def update_greenlet_frame(greenlet_id: int, frame: t.Union[FrameType, bool, None]) -> None:
     _greenlet_frames[greenlet_id] = frame
-    stack_v2.update_greenlet_frame(greenlet_id, frame)  # type: ignore[attr-defined]
+    stack_v2.update_greenlet_frame(greenlet_id, frame)
 
 
 def greenlet_tracer(event: str, args: t.Any) -> None:
@@ -106,7 +106,7 @@ def greenlet_tracer(event: str, args: t.Any) -> None:
 
 def untrack_greenlet(greenlet: _Greenlet) -> None:
     greenlet_id = thread.get_ident(greenlet)
-    stack_v2.untrack_greenlet(greenlet_id)  # type: ignore[attr-defined]
+    stack_v2.untrack_greenlet(greenlet_id)
     _greenlet_frames.pop(greenlet_id, None)
     _parent_greenlet_count.pop(greenlet_id, None)
     if (parent_id := _greenlet_parent_map.pop(greenlet_id, None)) is not None:
@@ -116,7 +116,7 @@ def untrack_greenlet(greenlet: _Greenlet) -> None:
 
 
 def link_greenlets(greenlet_id: int, parent_id: int) -> None:
-    stack_v2.link_greenlets(greenlet_id, parent_id)  # type: ignore[attr-defined]
+    stack_v2.link_greenlets(greenlet_id, parent_id)
     _parent_greenlet_count[parent_id] = _parent_greenlet_count.get(parent_id, 0) + 1
     _greenlet_parent_map[greenlet_id] = parent_id
 
@@ -159,7 +159,7 @@ def _(*args: t.Any, **kwargs: t.Any) -> _Greenlet:
     return _
 
 
-def joinall(greenlets: t.Iterable[_Greenlet], *args: t.Any, **kwargs: t.Any) -> None:
+def joinall(greenlets: t.Sequence[_Greenlet], *args: t.Any, **kwargs: t.Any) -> None:
     # This is a wrapper around gevent.joinall to track the greenlets
     # that are being joined.
     current_greenlet = gevent.getcurrent()