SNOW-1867961: from_json does not work for TimestampType if it contains tz_info

[AdamosX]

Relevant functions:

snowpark-python/src/snowflake/snowpark/types.py

Line 927 in 69c41a2

def _parse_datatype_json_value(json_value: Union[dict, str]) -> DataType:

snowpark-python/src/snowflake/snowpark/types.py

Line 209 in 69c41a2

def json_value(self) -> str:

[sfc-gh-sghosh]

Hello @AdamosX ,

Thanks for raising the issue. Could you clarify the requirement for the scenario and please share the code snippet as well?
Are you trying to convert JSON to custom python data types ?

Regards,
Sujan

[AdamosX]

Sure, I just read a table schema from a real table on snowflake and serialized it to json.
Then I wanted to deserialize it from json, and it failed.

The code below contains the simplified relevant parts:

from snowflake.snowpark.types import StructType, TimestampType, TimestampTimeZone, StructField
s = StructType(
        [
            StructField(
                "INSERT_TIMESTAMP",
                TimestampType(timezone=TimestampTimeZone.NTZ), #if I remove timezone, it works
                nullable=True,
            ),
        ]
    )
j = s.json_value()
sj = StructType.from_json(j) #does not work: ValueError: Cannot parse data type: timestamp_ntz

[sfc-gh-sghosh]

Thank you @AdamosX ,

We are able to reproduce the issue, and will work on eliminating it, and will update.

Regards,
Sujan