Skip to content

Commit 8007c27

Browse files
address comments
1 parent b82b55a commit 8007c27

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

src/snowflake/snowpark/modin/plugin/io/snow_io.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -399,7 +399,10 @@ def read_csv(
399399
if (
400400
filepath_or_buffer is not None
401401
and isinstance(filepath_or_buffer, str)
402-
and filepath_or_buffer.lower().startswith("s3")
402+
and any(
403+
filepath_or_buffer.lower().startswith(prefix)
404+
for prefix in ["s3://", "s3china://", "s3gov://"]
405+
)
403406
):
404407
session = get_active_session()
405408
temp_stage_name = random_name_for_temp_object(TempObjectType.STAGE)

tests/integ/modin/io/test_read_csv.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -840,6 +840,9 @@ def test_read_csv_dateparse_multiple_columns():
840840

841841

842842
def test_read_csv_s3():
843+
host = pd.session.connection.host
844+
if any(platform in host.split(".") for platform in ["gcp", "azure"]):
845+
pytest.mark.skip(reason="Skipping test for Azure and GCP deployment")
843846
with SqlCounter(query_count=9):
844847
df = pd.read_csv(
845848
"s3://sfquickstarts/frostbyte_tastybytes/analytics/menu_item_aggregate_v.csv"

0 commit comments

Comments
 (0)