Visual Studio不显示来自awswrangler的DataFrame弹出帮助

Question

我正在使用带有Microsoft Python扩展的VS代码。如果我创建一个Pandas dataframe并写出变量的名称VS Code，弹出各种帮助文本。但是，如果我有一个使用wr.athena.read_sql_query创建的变量，即使该变量是一个Pandas数据帧，我也得不到任何帮助文本。

有没有办法让VS代码意识到示例中的df2是一个熊猫DataFrame并获得帮助文本？

import boto3
import awswrangler as wr
import pandas as pd

df1 = pd.DataFrame({"a":[1]})
df2 = wr.athena.read_sql_query(sql="...", database="...")

Answer 1

如下所示添加键入提示：

df2: pd.DataFrame = wr.athena.read_sql_query(sql="...", database="...")

应该做这件事。