From 669f6bbf0e53c9356af3e938c79a6eea25e39459 Mon Sep 17 00:00:00 2001 From: Jonathan Shi Date: Tue, 16 Jun 2026 11:08:04 -0700 Subject: [PATCH 1/2] xfail ai_split tests --- tests/integ/test_dataframe_ai.py | 30 ++++++++++++++++++++++++------ 1 file changed, 24 insertions(+), 6 deletions(-) diff --git a/tests/integ/test_dataframe_ai.py b/tests/integ/test_dataframe_ai.py index dc6f280b60..23e3a5557b 100644 --- a/tests/integ/test_dataframe_ai.py +++ b/tests/integ/test_dataframe_ai.py @@ -2,7 +2,6 @@ # Copyright (c) 2012-2025 Snowflake Computing Inc. All rights reserved. # -import datetime import json import pytest from snowflake.snowpark.functions import ai_complete, col, lit, to_file @@ -16,11 +15,6 @@ "config.getoption('local_testing_mode', default=False)", reason="AI functions are not yet supported in local testing mode.", ), - pytest.mark.xfail( - datetime.date.today() <= datetime.date(2026, 6, 4), - reason="AI tests flaky due to infrastructure issues", - strict=False, - ), ] @@ -1294,6 +1288,10 @@ def test_dataframe_ai_count_tokens_error_handling(session): ) +@pytest.mark.xfail( + reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", + strict=True, +) def test_dataframe_ai_split_text_markdown_header_basic(session): """Test DataFrame.ai.split_text_markdown_header with basic usage.""" # Create a DataFrame with Markdown text @@ -1369,6 +1367,10 @@ def test_dataframe_ai_split_text_markdown_header_basic(session): ] +@pytest.mark.xfail( + reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", + strict=True, +) def test_dataframe_ai_split_text_markdown_header_default_output(session): """Test DataFrame.ai.split_text_markdown_header with default output column.""" df = session.create_dataframe( @@ -1401,6 +1403,10 @@ def test_dataframe_ai_split_text_markdown_header_error_handling(session): ) +@pytest.mark.xfail( + reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", + strict=True, +) def test_dataframe_ai_split_text_recursive_character_basic(session): """Test DataFrame.ai.split_text_recursive_character with basic usage.""" long_text = """This is a long document with multiple sentences. It contains various information @@ -1440,6 +1446,10 @@ def test_dataframe_ai_split_text_recursive_character_basic(session): ] +@pytest.mark.xfail( + reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", + strict=True, +) def test_dataframe_ai_split_text_recursive_character_markdown_format(session): """Test DataFrame.ai.split_text_recursive_character with markdown format.""" markdown_text = """# Main Title @@ -1497,6 +1507,10 @@ def hello(): ] +@pytest.mark.xfail( + reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", + strict=True, +) def test_dataframe_ai_split_text_recursive_character_custom_separators(session): """Test DataFrame.ai.split_text_recursive_character with custom separators.""" code_text = """def function_one(): @@ -1538,6 +1552,10 @@ def function_three(): ] +@pytest.mark.xfail( + reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", + strict=True, +) def test_dataframe_ai_split_text_recursive_character_default_output(session): """Test DataFrame.ai.split_text_recursive_character with default output column.""" df = session.create_dataframe( From 12bd1f7fcc17d51bdb85f7aadb21270a4bd594ec Mon Sep 17 00:00:00 2001 From: Jonathan Shi Date: Tue, 16 Jun 2026 14:09:03 -0700 Subject: [PATCH 2/2] xfail only on jenkins (preprod) --- tests/integ/test_dataframe_ai.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/integ/test_dataframe_ai.py b/tests/integ/test_dataframe_ai.py index 23e3a5557b..77aaab0b19 100644 --- a/tests/integ/test_dataframe_ai.py +++ b/tests/integ/test_dataframe_ai.py @@ -6,7 +6,7 @@ import pytest from snowflake.snowpark.functions import ai_complete, col, lit, to_file from snowflake.snowpark.row import Row -from tests.utils import TestFiles, Utils +from tests.utils import TestFiles, Utils, running_on_jenkins from snowflake.snowpark.exceptions import SnowparkSQLException @@ -1289,6 +1289,7 @@ def test_dataframe_ai_count_tokens_error_handling(session): @pytest.mark.xfail( + running_on_jenkins(), reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", strict=True, ) @@ -1368,6 +1369,7 @@ def test_dataframe_ai_split_text_markdown_header_basic(session): @pytest.mark.xfail( + running_on_jenkins(), reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", strict=True, ) @@ -1404,6 +1406,7 @@ def test_dataframe_ai_split_text_markdown_header_error_handling(session): @pytest.mark.xfail( + running_on_jenkins(), reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", strict=True, ) @@ -1447,6 +1450,7 @@ def test_dataframe_ai_split_text_recursive_character_basic(session): @pytest.mark.xfail( + running_on_jenkins(), reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", strict=True, ) @@ -1508,6 +1512,7 @@ def hello(): @pytest.mark.xfail( + running_on_jenkins(), reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", strict=True, ) @@ -1553,6 +1558,7 @@ def function_three(): @pytest.mark.xfail( + running_on_jenkins(), reason="SNOW-3664377: AI_SPLIT tests fail due to server-side anaconda repository validation error", strict=True, )