Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adjust Systems to use TensorTable and schema shapes #293

Merged
merged 5 commits into from
Mar 17, 2023

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
Transformed tensor dictionary
"""
candidate_ids = transformable[self._input_col]
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

merlin/systems/dag/ops/pytorch.py Outdated Show resolved Hide resolved
merlin/systems/dag/ops/pytorch.py Outdated Show resolved Hide resolved
merlin/systems/dag/ops/workflow.py Outdated Show resolved Hide resolved
merlin/systems/dag/ops/workflow.py Outdated Show resolved Hide resolved
tests/unit/systems/ops/faiss/test_executor.py Outdated Show resolved Hide resolved
@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