SNOW-1000455: Incorrect return type hint in DataFrameNaFunctions class

[TedCha]

1. What version of Python are you using?

Python 3.8.10 (tags/v3.8.10:3d8993a, May 3 2021, 11:48:03) [MSC v.1928 64 bit (AMD64)]

2. What operating system and processor architecture are you using?

Windows-10-10.0.19045-SP0

3. What are the component versions in the environment (pip freeze)?

asn1crypto==1.5.1
certifi==2023.11.17      
cffi==1.16.0
charset-normalizer==3.3.2
cloudpickle==2.2.1       
colorama==0.4.6
cryptography==41.0.7     
exceptiongroup==1.2.0    
filelock==3.13.1
idna==3.6
iniconfig==2.0.0
numpy==1.24.4
packaging==23.2
pandas==2.0.3
platformdirs==3.11.0
pluggy==1.3.0
pyarrow==14.0.2
pycparser==2.21
PyJWT==2.8.0
pyOpenSSL==23.3.0
pytest==7.4.4
python-dateutil==2.8.2
pytz==2023.3.post1
PyYAML==6.0.1
requests==2.31.0
six==1.16.0
snowflake-connector-python==3.6.0
snowflake-snowpark-python==1.11.1
sortedcontainers==2.4.0
tomli==2.0.1
tomlkit==0.12.3
typing_extensions==4.9.0
tzdata==2023.4
urllib3==1.26.18

4. What did you do?

When using any of the DataFrameNaFunctions methods, the return type will be Unknown.

import snowflake.snowpark as snowpark
from snowflake.snowpark.types import StructType, StructField, StringType

def main(session: snowpark.Session):

    schema = StructType([StructField("A", StringType())])
    df = session.create_dataframe([], schema) # snowflake.snowpark.dataframe.DataFrame
    
    df = df.na.drop() # Unknown

5. What did you expect to see?

The return type should be snowflake.snowpark.dataframe.DataFrame.

6. Can you set logging to DEBUG and collect the logs?

Don't think logs are applicable as issue is with type hinting but can provide screenshots: