Skip to content

Conversation

@karlhigley
Copy link
Contributor

@karlhigley karlhigley commented Mar 7, 2023

Depends on NVIDIA-Merlin/core#235

This also ended up depending on the shape and dict tuple -> separate dict keys changes in the models and dataloaders.

@karlhigley karlhigley added clean up chore Maintenance for the repository labels Mar 7, 2023
@karlhigley karlhigley added this to the Merlin 23.03 milestone Mar 7, 2023
@karlhigley karlhigley self-assigned this Mar 7, 2023
filter_ids = transformable[self._filter_out_col]

filtered_results = candidate_ids[~np.isin(candidate_ids, filter_ids)]
filtered_results = candidate_ids.values[~np.isin(candidate_ids.values, filter_ids.values)]
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is a workaround since TensorColumn doesn't support indexing with a boolean array yet

@karlhigley karlhigley requested a review from jperez999 March 8, 2023 00:51
@karlhigley karlhigley marked this pull request as ready for review March 8, 2023 14:56
@karlhigley karlhigley merged commit d64d913 into NVIDIA-Merlin:main Mar 17, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

chore Maintenance for the repository clean up

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants