Support list comprehension for TypeVarTuple

[zen-xu]

Imagine there's a scenario, I need to accept a bunch of different type objects, and then return these objects boxed in a tuple of a generic type. Currently, this can only be achieved by manually writing overload, is there a more pythonic solution?

from typing import Generic
from typing import TypeVar
from typing import overload


_T = TypeVar("_T")
_T0 = TypeVar("_T0")
_T1 = TypeVar("_T1")
_T2 = TypeVar("_T2")


class ObjectRef(Generic[_T]): ...


@overload
def put(items: tuple[_T0]) -> tuple[ObjectRef[_T0]]: ...


@overload
def put(items: tuple[_T0, _T1]) -> tuple[ObjectRef[_T0], ObjectRef[_T1]]: ...


@overload
def put(
    items: tuple[_T0, _T1, _T2],
) -> tuple[ObjectRef[_T0], ObjectRef[_T1], ObjectRef[_T2]]: ...


@overload
def get(refs: tuple[ObjectRef[_T0]]) -> tuple[_T0]: ...


@overload
def get(refs: tuple[ObjectRef[_T0], ObjectRef[_T1]]) -> tuple[_T0, _T1]: ...


@overload
def get(
    refs: tuple[ObjectRef[_T0], ObjectRef[_T1], ObjectRef[_T2]],
) -> tuple[_T0, _T1, _T2]: ...

[zen-xu]

If TypeVarTuple supports list comprehension, that would be great

from typing import Generic
from typing import TypeVar
from typing import TypeVarTuple


_T = TypeVar("_T")
_Ts = TypeVarTuple("_Ts")


class ObjectRef(Generic[_T]): ...

def put(items: tuple[*_Ts]) -> tuple[ObjectRef[_T] for _T in _Ts]: ...

# backport to old python version
# def put(items: tuple[*_Ts]) -> tuple["ObjectRef[_T] for _T in _Ts"]: ...


def get(refs: tuple[*_Ts]) -> tuple[_T for ObjectRef[_T] in _Ts]: ...

We can do more things like.

from typing import Any
from typing import Callable
from typing import Generic
from typing import TypeVar
from typing import TypeVarTuple


_Callable_co = TypeVar("_Callable_co", covariant=True, bound=Callable)
_Ts = TypeVarTuple("_Ts")
_T = TypeVar("_T")


class ObjectRef(Generic[_T]): ...


class RemoteCallable(Generic[_Callable_co]):
    def __call__(self: RemoteCallable[Callable[[*_Ts], Any]], *args: *tuple[ObjectRef[_T] for _T in _Ts]): ...

For a more realistic scenario, you can refer to https://github.com/zen-xu/sunray/blob/main/sunray/_internal/callable.py

[carljm]

I think this is a duplicate of #1216