feat(optimizer)!: Annotate type for snowflake STRTOK_TO_ARRAY function (#5994)

* feat(optimizer): Annotate type for snowflake STRTOK_TO_ARRAY function

* fix: Updated tests

Author: fivetran-amrutabhimsenayachit

sqlglot/dialects/snowflake.py

@@ -591,6 +591,7 @@ class Snowflake(Dialect):
         exp.DataType.Type.ARRAY: {
             exp.Split,
             exp.RegexpExtractAll,
+            exp.StringToArray,
         },
         exp.DataType.Type.OBJECT: {
             exp.ParseUrl,

tests/fixtures/optimizer/annotate_functions.sql

@@ -2187,6 +2187,10 @@ BOOLEAN;
 STARTSWITH(tbl.bin_col, NULL);
 BOOLEAN;
 
+# dialect: snowflake
+STRTOK_TO_ARRAY('a,b,c', ',');
+ARRAY;
+
 # dialect: snowflake
 SUBSTR('hello world', 1, 5);
 VARCHAR;