Skip to content

Commit ef3cda2

Browse files
fix errors
1 parent b44816c commit ef3cda2

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

src/snowflake/snowpark/modin/plugin/docstrings/series_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2642,7 +2642,7 @@ def seconds():
26422642
0 1
26432643
1 2
26442644
2 3
2645-
dtype: int64
2645+
dtype: int8
26462646
26472647
For TimedeltaIndex:
26482648
@@ -2702,7 +2702,7 @@ def microseconds():
27022702
0 1
27032703
1 2
27042704
2 3
2705-
dtype: int64
2705+
dtype: int8
27062706
27072707
For TimedeltaIndex:
27082708
@@ -2734,7 +2734,7 @@ def nanoseconds():
27342734
0 1
27352735
1 2
27362736
2 3
2737-
dtype: int64
2737+
dtype: int8
27382738
27392739
For TimedeltaIndex:
27402740

tests/integ/modin/test_faster_pandas.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -123,9 +123,9 @@ def test_read_filter_join_on_index(session):
123123
)
124124

125125

126-
@sql_count_checker(query_count=3)
127-
def test_read_filter_groupby_agg(session):
128-
# test a chain of operations that are not fully supported in faster pandas
126+
@sql_count_checker(query_count=3, join_count=2)
127+
def test_read_filter_iloc_index(session):
128+
# test a chain of operations that are not yet fully supported in faster pandas
129129

130130
# create tables
131131
table_name = Utils.random_name_for_temp_object(TempObjectType.TABLE)
@@ -135,19 +135,19 @@ def test_read_filter_groupby_agg(session):
135135

136136
# create snow dataframes
137137
df = pd.read_snowflake(table_name)
138-
snow_result = df[df["B"] > 11].groupby("A").min()
138+
snow_result = df.iloc[[0], :]
139139

140140
# verify that the input dataframe has a populated relaxed query compiler
141141
assert df._query_compiler._relaxed_query_compiler is not None
142142
assert df._query_compiler._relaxed_query_compiler._dummy_row_pos_mode is True
143143
# verify that the output dataframe has an empty relaxed query compiler
144-
# because groupby() and min() are not supported in faster pandas yet
144+
# because iloc for index is not supported in faster pandas yet
145145
assert snow_result._query_compiler._relaxed_query_compiler is None
146146
assert snow_result._query_compiler._dummy_row_pos_mode is False
147147

148148
# create pandas dataframes
149149
native_df = df.to_pandas()
150-
native_result = native_df[native_df["B"] > 11].groupby("A").min()
150+
native_result = native_df.iloc[[0], :]
151151

152152
# compare results
153153
assert_frame_equal(snow_result, native_result)

0 commit comments

Comments
 (0)