使用函数的参数注释python函数的返回类型

Question

我目前正在编写如下代码：

def doSomethingWithClass(className, param):
    pass # some logic that returns an object of type className

例如，调用doSomethingWithClass(ClassA, 3)的实例将返回ClassA的实例

我正在尝试为此函数编写类型注释，并且我想表示返回类型是作为第一个参数传入的类。从语法上讲，我正在寻找这样的东西：

def doSomethingWithClass(className: MetaClass, param: int) -> className
    pass # same logic here

但是，这不起作用，因为参数className不在返回类型注释所在的函数范围之外。有写我想做的事情的方法吗？