Skip to content

Commit

Permalink
Add nvtx annotations for Series and Index (#10374)
Browse files Browse the repository at this point in the history
This PR adds some missing `nvtx` annotations for `Series` and `Index`.

Authors:
  - GALI PREM SAGAR (https://github.com/galipremsagar)

Approvers:
  - Vyas Ramasubramani (https://github.com/vyasr)

URL: #10374
  • Loading branch information
galipremsagar authored Mar 3, 2022
1 parent 6bcfc10 commit b5337d7
Show file tree
Hide file tree
Showing 5 changed files with 485 additions and 97 deletions.
6 changes: 3 additions & 3 deletions python/cudf/cudf/core/frame.py
Original file line number Diff line number Diff line change
Expand Up @@ -6921,12 +6921,12 @@ def _drop_rows_by_labels(
raise KeyError("One or more values not found in axis")

key_df = cudf.DataFrame(index=labels)
if isinstance(obj, cudf.Series):
if isinstance(obj, cudf.DataFrame):
return obj.join(key_df, how="leftanti")
else:
res = obj.to_frame(name="tmp").join(key_df, how="leftanti")["tmp"]
res.name = obj.name
return res
else:
return obj.join(key_df, how="leftanti")


def _apply_inverse_column(col: ColumnBase) -> ColumnBase:
Expand Down
Loading

0 comments on commit b5337d7

Please sign in to comment.