SNOW-896744: Allow creating dataframes using tuples as schema (#1025)

* SNOW-896744: Allow creating dataframes using tuples as schema

* use iterable instead of just tuple

* changelog updates

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Release History
 
+## 1.8.0 (TBD)
+
+### New Features
+
+- Accept `Iterable` objects input for `schema` when creating dataframes using `Session.create_dataframe`.
+
 ## 1.7.0 (2023-08-28)
 
 ### New Features

src/snowflake/snowpark/session.py

@@ -1865,7 +1865,7 @@ def write_pandas(
     def create_dataframe(
         self,
         data: Union[List, Tuple, "pandas.DataFrame"],
-        schema: Optional[Union[StructType, List[str]]] = None,
+        schema: Optional[Union[StructType, Iterable[str]]] = None,
     ) -> DataFrame:
         """Creates a new DataFrame containing the specified values from the local data.
 
@@ -1961,8 +1961,8 @@ def create_dataframe(
         else:
             if not data:
                 raise ValueError("Cannot infer schema from empty data")
-            if isinstance(schema, list):
-                names = schema
+            if isinstance(schema, Iterable):
+                names = list(schema)
             new_schema = reduce(
                 merge_type,
                 (infer_schema(row, names) for row in data),

tests/integ/test_dataframe.py

@@ -2855,6 +2855,13 @@ def test_create_dataframe_special_char_column_name(session):
     Utils.check_answer(df2, [Row(1, 2, 3), Row(1, 2, 3)])
 
 
+def test_create_dataframe_with_tuple_schema(session):
+    df = session.create_dataframe(
+        [(20000101, 1, "x"), (20000101, 2, "y")], schema=("TIME", "ID", "V2")
+    )
+    Utils.check_answer(df, [Row(20000101, 1, "x"), Row(20000101, 2, "y")])
+
+
 def test_df_join_suffix(session):
     df1 = session.create_dataframe([[1, 1, "1"], [2, 2, "3"]]).to_df(["a", "b", "c"])
     df2 = session.create_dataframe([[1, 1, "1"], [2, 3, "5"]]).to_df(["a", "b", "c"])