diff --git a/setup.py b/setup.py index 000bdfa338..a3e3699e8f 100644 --- a/setup.py +++ b/setup.py @@ -66,6 +66,7 @@ "snowflake.core>=1.0.0, <2", # Catalog "psutil", # testing for telemetry "lxml", # used in XML reader unit tests + "pyarrow", # used in dataframe reader tests ] MODIN_DEVELOPMENT_REQUIREMENTS = [ # Snowpark pandas 3rd party library testing. Cap the scipy version because diff --git a/tests/integ/compiler/test_query_generator.py b/tests/integ/compiler/test_query_generator.py index e3cdf38889..e7c1e1d181 100644 --- a/tests/integ/compiler/test_query_generator.py +++ b/tests/integ/compiler/test_query_generator.py @@ -9,7 +9,13 @@ import os import re -import pandas +try: + import pandas + + is_pandas_available = True +except ImportError: + is_pandas_available = False + import pytest import snowflake.snowpark._internal.analyzer.snowflake_plan as snowflake_plan @@ -568,6 +574,7 @@ def test_select_alias_identity(session): ) +@pytest.mark.skipif(not is_pandas_available, reason="pandas is required") def test_disambiguate_skips_quoted_alias(session): # SNOW-3176017: This tests a previous regression in a SnowML pipeline where alias optimization # incorrectly removed an alias from """col_0""" (triple-quoted in SQL) to "col_0" (single-quoted). diff --git a/tests/notebooks/modin/TimeSeriesTesting.ipynb b/tests/notebooks/modin/TimeSeriesTesting.ipynb index 8fc305a4fc..ae791ddf03 100644 --- a/tests/notebooks/modin/TimeSeriesTesting.ipynb +++ b/tests/notebooks/modin/TimeSeriesTesting.ipynb @@ -346,7 +346,9 @@ } ], "source": [ - "ts.resample(\"2h\").mean()" + "# Skipped due to modin bug with pandas 2.3.3: https://github.com/modin-project/modin/issues/7697", + "# Cannot install lower versions of pandas 2.3 due to https://github.com/pandas-dev/pandas/issues/65213", + "# ts.resample(\"2h\").mean()" ] }, { diff --git a/tests/notebooks/test_requirements.txt b/tests/notebooks/test_requirements.txt index 3672bcd445..c344a5574e 100644 --- a/tests/notebooks/test_requirements.txt +++ b/tests/notebooks/test_requirements.txt @@ -1,4 +1,4 @@ -numpy==1.23.5 +numpy pytest<8.0.0 nbmake matplotlib