Digite dicas com classes definidas pelo usuário

Não foi possível encontrar uma resposta definitiva. Eu quero fazer uma dica de tipo para uma função e o tipo sendo alguma classe personalizada que eu defini, chamei-a CustomClass().

E então digamos que em alguma função, chame-a FuncA(arg), eu tenho um argumento chamado arg. A maneira correta de digitar a dica FuncAseria:

def FuncA(arg: CustomClass):

Ou seria:

def FuncA(Arg:Type[CustomClass]):?

O primeiro está correto , se argaceita uma instância deCustomClass :

def FuncA(arg: CustomClass):
    #     ^ instance of CustomClass

Caso queira a própria classe CustomClass(ou um subtipo) , você deve escrever:

from typing import Type  # you have to import Type

def FuncA(arg: Type[CustomClass]):
    #     ^ CustomClass (class object) itself

Como está escrito na documentação sobre Digitação :

class typing.Type(Generic[CT_co])

Uma variável anotada com Cpode aceitar um valor do tipo C. Em contraste, uma variável anotada com Type[C]pode aceitar valores que são classes em si - especificamente, ela aceitará o objeto de classe deC .

A documentação inclui um exemplo com a intclasse:

a = 3         # Has type 'int'
b = int       # Has type 'Type[int]'
c = type(a)   # Also has type 'Type[int]'