From e47d1a5cfe167dd2db7d42663502fcf4001bdea6 Mon Sep 17 00:00:00 2001 From: Naresh Kumar Date: Thu, 29 Aug 2024 17:30:15 -0700 Subject: [PATCH] add null to test data points --- .../snowpark/modin/plugin/compiler/snowflake_query_compiler.py | 2 +- tests/integ/modin/index/test_timedelta_index_methods.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/snowflake/snowpark/modin/plugin/compiler/snowflake_query_compiler.py b/src/snowflake/snowpark/modin/plugin/compiler/snowflake_query_compiler.py index 83aece29058..c7ba0180fd1 100644 --- a/src/snowflake/snowpark/modin/plugin/compiler/snowflake_query_compiler.py +++ b/src/snowflake/snowpark/modin/plugin/compiler/snowflake_query_compiler.py @@ -17543,7 +17543,7 @@ def timedelta_property( ) # pragma: no cover raise ErrorMessage.not_implemented( f"Snowpark pandas doesn't yet support the property '{class_prefix}.{property_name}'" - ) + ) # pragma: no cover return SnowflakeQueryCompiler( self._modin_frame.apply_snowpark_function_to_columns(func, include_index) ) diff --git a/tests/integ/modin/index/test_timedelta_index_methods.py b/tests/integ/modin/index/test_timedelta_index_methods.py index c68ab9653fa..646bd5ee983 100644 --- a/tests/integ/modin/index/test_timedelta_index_methods.py +++ b/tests/integ/modin/index/test_timedelta_index_methods.py @@ -68,7 +68,7 @@ def test_property_not_implemented(property): @sql_count_checker(query_count=1) def test_timedelta_index_properties(attr): native_index = native_pd.TimedeltaIndex( - ["1d", "1h", "60s", "1s", "800ms", "5us", "6ns", "1d 3s", "9m 15s 8us"] + ["1d", "1h", "60s", "1s", "800ms", "5us", "6ns", "1d 3s", "9m 15s 8us", None] ) snow_index = pd.Index(native_index) assert_index_equal(