These templates are a set of minimal examples that are quick and easy to run and customize.
Although the templates may include some machine learning framework-specific code, the individual code blocks are meant to be swapped in with your own application logic. The templates just serve as skeletons that showcase popular applications of Ray.
Coming soon...
To add a template:
-
Add your template as a directory somewhere in
doc/source/templates.For example:
ray/ doc/source/templates/ <name-of-your-template>/ README.md <name-of-your-template>.ipynb requirements.txt (Optional) templates.yamlYour template does not need to be a Jupyter notebook. It can also be presented as a Python script with
READMEinstructions of how to run. -
Add a release test for the template in
release/release_tests.yaml(for both AWS and GCE). For Data tests, userelease/release_data_tests.yamlinstead.See the section on workspace templates for an example. Note that the cluster env and compute config are a little different for release tests. Use the files in the
doc/source/templates/testing/releasefolder.The release test compute configs contain placeholders for regions and cloud ids that our CI infra will fill in. The cluster env builds a nightly docker image with all the required dependencies.
-
Add an entry to
doc/source/templates/templates.yamlthat links to your template.See the top of the
templates.yamlfile for something to copy-paste and fill in your own values.When you specify the template's compute config, see
doc/source/templates/configsfor shared configs. You can also create custom compute configs (of the same format as these shared ones).For handling dependencies:
-
If your template requires any special dependencies that are not included in a base image that you chose, be sure to list and provide instructions to install the necessary dependencies within the notebook. See
02_many_model_trainingfor an example. -
If your template requires a custom docker image, be sure to mention this in the
READMEand link the docker image URL somewhere. See03_serving_stable_diffusionfor an example.
-
-
Run a validation script on
templates.yamlto make sure that the paths you specified are all valid and all yamls are properly formatted.Note: This will also run in CI, but you can check quickly by running the validation script.
$ python doc/source/templates/testing/validate.py Success! -
Success! Your template is ready for review.