from __future__ import annotations

from dataclasses import dataclass
from functools import partial
from typing import Any, TypeVar, ParamSpec, Generic, Protocol, Callable


_T = TypeVar("_T")
_P = ParamSpec("_P")


@dataclass(kw_only=True, repr=True)
class Task(Generic[_P, _T]):
    name: str
    function: Callable[_P, _T]


class _TaskRegistrator(Protocol):
    def __call__(self, func: Callable[_P, _T], /, *, name: str = ...) -> Task[_P, _T]:
        ...


class TaskQueue:
    def _register_task(self, func: Callable[_P, _T], *, name: str) -> Task[_P, _T]:
        # Actual implementation omitted
        return Task(name=name, function=func)

    def task(self, name: str) -> _TaskRegistrator:
        return partial(self._register_task, name=name)  # E: see below (L29)
        
    def task2(self, name: str) -> Callable[[Callable[_P, _T]], Task[_P, _T]]:
        return partial(self._register_task, name=name)  # E: see below (L32)


# And now we see that the problem is even unrelated to `self`:

def _register_task(func: Callable[_P, _T], *, name: str) -> Task[_P, _T]:
    # Actual implementation omitted
    return Task(name=name, function=func)

def task(name: str) -> _TaskRegistrator:
    return partial(_register_task, name=name)  # E: see below (L42)