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

[rllib] Enhancements to Input API for customizing offline datasets #16957

Merged
merged 26 commits into from Jul 10, 2021

Conversation

juliusfrost
Copy link
Member

@juliusfrost juliusfrost commented Jul 8, 2021

Why are these changes needed?

This PR makes useful enhancements to the Input API so that it is much easier to customize offline datasets.
It implements the following:

  • input registry
  • input by class path
  • custom input example
  • input configuration
  • documentation

Related issue number

Checks

  • I've run scripts/format.sh to lint the changes in this PR.
  • I've included any doc changes needed for https://docs.ray.io/en/master/.
  • I've made sure the tests are passing. Note that there might be a few flaky tests, see the recent failures at https://flakey-tests.ray.io/
  • Testing Strategy
    • Unit tests
    • Release tests
    • This PR is not tested :(

I would love feedback from @richardliaw @sven1977 @michaelzhiluo

@richardliaw richardliaw self-assigned this Jul 8, 2021
Copy link
Member

@stefanbschneider stefanbschneider left a comment

Choose a reason for hiding this comment

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

Hi, thanks for the contribution! :)

The changes look good to me (I just had a quick look). Did you already run format.sh? And do you still plan on writing some test(s) to test the new functionality?

doc/source/rllib-offline.rst Show resolved Hide resolved
rllib/examples/custom_input_api.py Show resolved Hide resolved
@juliusfrost juliusfrost marked this pull request as ready for review July 8, 2021 21:49
@juliusfrost
Copy link
Member Author

Hi, thanks for the contribution! :)

The changes look good to me (I just had a quick look). Did you already run format.sh? And do you still plan on writing some test(s) to test the new functionality?

Hi, thanks for the review! I don't have a machine on hand to run format.sh. Also, I'm not sure how to create tests for the offline API, are there any examples or tests I can modify?

@richardliaw
Copy link
Contributor

richardliaw commented Jul 8, 2021 via email

@richardliaw
Copy link
Contributor

richardliaw commented Jul 8, 2021 via email

@juliusfrost
Copy link
Member Author

Added tests. Should be ready to go if it passes

juliusfrost and others added 2 commits July 9, 2021 02:10
Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
@richardliaw
Copy link
Contributor

richardliaw commented Jul 10, 2021 via email

@juliusfrost
Copy link
Member Author

Tests are passing!

Copy link
Contributor

@richardliaw richardliaw left a comment

Choose a reason for hiding this comment

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

nice!

@richardliaw
Copy link
Contributor

@michaelzhiluo @sven1977 I'll merge this for now; please take a look when convenient!

@richardliaw richardliaw merged commit a88b217 into ray-project:master Jul 10, 2021
@richardliaw
Copy link
Contributor

@juliusfrost great work here :)

@juliusfrost juliusfrost deleted the rllib-input-api branch July 10, 2021 22:40
jiaodong pushed a commit that referenced this pull request Jul 11, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants