Как аннотировать, что функция создает класс данных? ⇐ Python

[Anonymous]

Предположим, вы хотите обернуть декоратор класса данных следующим образом:
from dataclasses import dataclass

def something_else(klass):
return klass

def my_dataclass(klass):
return something_else(dataclass(klass))

Как следует аннотировать my_dataclass и/или Something_else, чтобы указать, что тип возвращаемого значения является классом данных?
См. следующий пример того, как встроенная функция @dataclass работает, а пользовательский @my_dataclass — нет:

@dataclass
class TestA:
a: int
b: str

TestA(0, "") # fine

@my_dataclass
class TestB:
a: int
b: str

TestB(0, "") # error: Too many arguments for "TestB" (from mypy)


Подробнее здесь: https://stackoverflow.com/questions/715 ... -dataclass