Thank you for your interest in contributing to SkyPilot! We welcome and value all contributions to the project, including but not limited to:
- Bug reports and discussions
- Pull requests for bug fixes and new features
- Test cases to make the codebase more robust
- Examples
- Documentation
- Tutorials, blog posts and talks on SkyPilot
We use GitHub to track issues and features. For new contributors, we recommend looking at issues labeled "good first issue".
# SkyPilot requires python >= 3.6.
# You can just install the dependencies for
# certain clouds, e.g., ".[aws,azure,gcp,lambda]"
pip install -e ".[all]"
pip install -r requirements-dev.txt
To run smoke tests (NOTE: Running all smoke tests launches ~20 clusters):
# Run all tests except for AWS and Lambda Cloud
pytest tests/test_smoke.py
# Terminate a test's cluster even if the test failed (default is to keep it around for debugging)
pytest tests/test_smoke.py --terminate-on-failure
# Re-run last failed tests
pytest --lf
# Run one of the smoke tests
pytest tests/test_smoke.py::test_minimal
# Only run managed spot tests
pytest tests/test_smoke.py --managed-spot
# Only run test for AWS + generic tests
pytest tests/test_smoke.py --aws
# Change cloud for generic tests to aws
pytest tests/test_smoke.py --generic-cloud aws
For profiling code, use:
pip install tuna # Tuna is used for visualization of profiling data.
python3 -m cProfile -o sky.prof -m sky.cli status # Or some other command
tuna sky.prof
- Fork the SkyPilot repository and create a new branch for your changes.
- If relevant, add tests for your changes. For changes that touch the core system, run the smoke tests and ensure they pass.
- Follow the Google style guide.
- Ensure code is properly formatted by running
format.sh
. - Push your changes to your fork and open a pull request in the SkyPilot repository.
- In the PR description, write a
Tested:
section to describe relevant tests performed.
These are suggestions, not strict rules to follow. When in doubt, follow the style guide.
- Use
TODO(author_name)
/FIXME(author_name)
instead of blankTODO/FIXME
. This is critical for tracking down issues. You can write TODOs with your name and assign it to others (on github) if it is someone else's issue. - Delete your branch after merging it. This keeps the repo clean and faster to sync.
- Use an exception if this is an error. Only use
assert
for debugging or proof-checking purposes. This is because exception messages usually contain more information. - Use modern python features and styles that increases code quality.
- Use f-string instead of
.format()
for short expressions to increase readability. - Use
class MyClass:
instead ofclass MyClass(object):
. The later one was a workaround for python2.x. - Use
abc
module for abstract classes to ensure all abstract methods are implemented. - Use python typing. But you should not import external objects just for typing. Instead, import typing-only external objects under
if typing.TYPE_CHECKING:
.
- Use f-string instead of
export SKYPILOT_DISABLE_USAGE_COLLECTION=1
to disable usage logging.export SKYPILOT_DEBUG=1
to show debugging logs (use logging.DEBUG level).export SKYPILOT_MINIMIZE_LOGGING=1
to minimize logging. Useful when trying to avoid multiple lines of output, such as for demos.