tensorflow / recommenders-addons

Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.

Apache License 2.0

596 stars 136 forks source link

[Feat]combine cpu and gpu wheel #456

Closed jq closed 2 months ago

jq commented 3 months ago

Description

Brief Description of the PR: combine cpu and gpu wheel so that you only need pip install tensorflow-recommenders-addons Fixes # (issue)

Type of change

[ ] Bug fix
[ ] New Tutorial
[ ] Updated or additional documentation
[ ] Additional Testing
[ ] New Feature

Checklist:

[ ] I've properly formatted my code according to the guidelines
- [ ] By running yapf
- [ ] By running clang-format
[ ] This PR addresses an already submitted issue for TensorFlow Recommenders-Addons
[ ] I have made corresponding changes to the documentation
[ ] I have added tests that prove my fix is effective or that my feature works

How Has This Been Tested?

If you're adding a bugfix or new feature please describe the tests that you ran to verify your changes:

rhdong commented 2 months ago

Hey @MoFHeka pls review it when you feel convenient, basically, it's LGTM.