Escriba sugerencias con clases definidas por el usuario

Parece que no pude encontrar una respuesta definitiva. Quiero hacer una sugerencia de tipo para una función y el tipo es una clase personalizada que he definido, la llamé CustomClass().

Y luego digamos que en alguna función, llámala FuncA(arg), tengo un argumento llamado arg. ¿Sería la forma correcta de escribir sugerencia FuncA:

def FuncA(arg: CustomClass):

O sería:

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

Lo primero es correcto , si argacepta una instancia deCustomClass :

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

En caso de que desee la clase en CustomClasssí (o un subtipo) , debe escribir:

from typing import Type  # you have to import Type

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

Como está escrito en la documentación sobre mecanografía :

class typing.Type(Generic[CT_co])

Una variable anotada con Cpuede aceptar un valor de tipo C. Por el contrario, una variable anotada con Type[C]puede aceptar valores que son clases en sí mismas ; específicamente, aceptará el objeto de clase deC .

La documentación incluye un ejemplo con la intclase:

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