From 30a77e71fcec3c1494f100373d943c208f15bbe6 Mon Sep 17 00:00:00 2001
From: Jianzhun Du <68252326+sfc-gh-jdu@users.noreply.github.com>
Date: Thu, 18 Apr 2024 11:16:24 -0700
Subject: [PATCH] SNOW-1227759: Fix the type annotation of save_as_table
 (#1395)

---
 src/snowflake/snowpark/dataframe_writer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/snowflake/snowpark/dataframe_writer.py b/src/snowflake/snowpark/dataframe_writer.py
index 1bb0b90e9e5..b30c08f40c0 100644
--- a/src/snowflake/snowpark/dataframe_writer.py
+++ b/src/snowflake/snowpark/dataframe_writer.py
@@ -87,7 +87,7 @@ def save_as_table(
         column_order: str = "index",
         create_temp_table: bool = False,
         table_type: Literal["", "temp", "temporary", "transient"] = "",
-        clustering_keys: Iterable[Column],
+        clustering_keys: Optional[Iterable[ColumnOrName]] = None,
         statement_params: Optional[Dict[str, str]] = None,
         block: bool = True,
     ) -> None:
@@ -102,7 +102,7 @@ def save_as_table(
         column_order: str = "index",
         create_temp_table: bool = False,
         table_type: Literal["", "temp", "temporary", "transient"] = "",
-        clustering_keys: Iterable[Column],
+        clustering_keys: Optional[Iterable[ColumnOrName]] = None,
         statement_params: Optional[Dict[str, str]] = None,
         block: bool = False,
     ) -> AsyncJob: