Generic Type Assumption from enum key

[InvincibleRMC]

Is there any way to type annotate so a is of type Test[int] instead of Test[Any]?

from typing import overload, TypeVar, Literal
T = TypeVar('T')
from enum import Enum

class TypeEnum(Enum):
    INT = 0
    BOOL = 1

class Test(Generic[T]):


    @overload
    def __init__(self, foo: Literal[TypeEnum.INT], value: int | None) -> None:...

    @overload
    def __init__(self, foo: None, value: T) -> None: ...

    def __init__(self, foo, value):
        self.value = value

    @property
    def value(self) -> T:
        return self._value
    
    @value.setter
    def value(self, value: T) -> None:
        self._value = value


a = Test(TypeEnum.INT, None)
reveal_type(a)

b = Test(None, 3)
reveal_type(b)

[erictraut]

Yes, you can add a type annotation for the self parameter in the first overload. Pyright playground link.

For details, refer to this section of the typing spec.

[InvincibleRMC]

Ah that makes. Thanks so much.