Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -113,6 +113,7 @@
- `to_datetime`
- `drop`
- `invert`
- `duplicated`
- Reuse row count from the relaxed query compiler in `get_axis_len`.

#### Bug Fixes
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17034,6 +17034,26 @@ def duplicated(
self,
subset: Union[Hashable, Sequence[Hashable]] = None,
keep: DropKeep = "first",
) -> "SnowflakeQueryCompiler":
"""
Wrapper around _duplicated_internal to be supported in faster pandas.
"""
relaxed_query_compiler = None
if self._relaxed_query_compiler is not None:
relaxed_query_compiler = self._relaxed_query_compiler._duplicated_internal(
subset=subset,
keep=keep,
)
qc = self._duplicated_internal(
subset=subset,
keep=keep,
)
return self._maybe_set_relaxed_qc(qc, relaxed_query_compiler)

def _duplicated_internal(
self,
subset: Union[Hashable, Sequence[Hashable]] = None,
keep: DropKeep = "first",
) -> "SnowflakeQueryCompiler":
"""
Return boolean Series denoting duplicate rows.
Expand Down
29 changes: 29 additions & 0 deletions tests/integ/modin/test_faster_pandas.py
Original file line number Diff line number Diff line change
Expand Up @@ -223,6 +223,35 @@ def test_drop(session):
assert_frame_equal(snow_result, native_result)


@sql_count_checker(query_count=3, join_count=1)
def test_duplicated(session):
# create tables
table_name = Utils.random_name_for_temp_object(TempObjectType.TABLE)
session.create_dataframe(
native_pd.DataFrame([[2, 12], [2, 12], [3, 13]], columns=["A", "B"])
).write.save_as_table(table_name, table_type="temp")

# create snow dataframes
df = pd.read_snowflake(table_name)
snow_result = df.duplicated()

# verify that the input dataframe has a populated relaxed query compiler
assert df._query_compiler._relaxed_query_compiler is not None
assert df._query_compiler._relaxed_query_compiler._dummy_row_pos_mode is True
# verify that the output dataframe also has a populated relaxed query compiler
assert snow_result._query_compiler._relaxed_query_compiler is not None
assert (
snow_result._query_compiler._relaxed_query_compiler._dummy_row_pos_mode is True
)

# create pandas dataframes
native_df = df.to_pandas()
native_result = native_df.duplicated()

# compare results
assert_series_equal(snow_result, native_result)


@sql_count_checker(query_count=3)
def test_invert(session):
# create tables
Expand Down
Loading