如何在python3中键入-ornotate tensorflow.Session（）

Question

我试图为tf.Session()的参数注释类型。

在命令行上，type(tf.Session())就像这样：

>>> type(tf.Session())
>>> <class 'tensorflow.python.client.session.Session'>

然后，我试着像这样注释。

def func_with_sess_arg(image: np.ndarray,
                       sess: tf.python.client.session.Session):
                       ...

但它提出了

Traceback (most recent call last):
File "func_with_sess_arg.py", line 13, in <module>
    sess: tf.python.client.session.Session):
AttributeError: 'module' object has no attribute 'python'

如何正确注释tf.Session（）的参数类型？

Answer 1

输入

def func_with_sess_arg(image: np.ndarray,
                       sess: tf.Session):
                       ...