Чтобы аннотировать тип самого класса, используйте Type
.
from typing import Type
class AClass(object):
pass
def a_function(a_string: str, a_class: Type[AClass]) -> None:
pass
Поскольку класс является типом, имя класса может быть самой аннотацией:
def a_class_factory(a_class: Type[AClass], *args, **kwargs) -> AClass:
return a_class(*args, **kwargs)